Skip to content

Commit 8c9dc05

Browse files
author
Vincent Moens
committed
[CI] Upgrade to cuda 12.8
ghstack-source-id: e0ad7d6 Pull Request resolved: #2820
1 parent 8dd1be7 commit 8c9dc05

File tree

29 files changed

+114
-114
lines changed

29 files changed

+114
-114
lines changed

‎.github/unittest/linux_libs/scripts_ataridqn/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,18 +26,18 @@ fi
2626
# submodules
2727
git submodule sync && git submodule update --init --recursive
2828

29-
printf "Installing PyTorch with cu124"
29+
printf "Installing PyTorch with cu128"
3030
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3131
if [ "${CU_VERSION:-}" == cpu ] ; then
3232
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
3333
else
34-
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 -U
34+
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3535
fi
3636
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3737
if [ "${CU_VERSION:-}" == cpu ] ; then
3838
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
3939
else
40-
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124
40+
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128
4141
fi
4242
else
4343
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_brax/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,18 +25,18 @@ fi
2525
# submodules
2626
git submodule sync && git submodule update --init --recursive
2727

28-
printf "Installing PyTorch with cu124"
28+
printf "Installing PyTorch with cu128"
2929
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3030
if [ "${CU_VERSION:-}" == cpu ] ; then
3131
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
3232
else
33-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
33+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3434
fi
3535
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3636
if [ "${CU_VERSION:-}" == cpu ] ; then
3737
pip3 install torch --index-url https://download.pytorch.org/whl/cpu -U
3838
else
39-
pip3 install torch --index-url https://download.pytorch.org/whl/cu124
39+
pip3 install torch --index-url https://download.pytorch.org/whl/cu128
4040
fi
4141
else
4242
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_chess/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,18 +41,18 @@ fi
4141
# submodules
4242
git submodule sync && git submodule update --init --recursive
4343

44-
printf "Installing PyTorch with cu124"
44+
printf "Installing PyTorch with cu128"
4545
if [[ "$TORCH_VERSION" == "nightly" ]]; then
4646
if [ "${CU_VERSION:-}" == cpu ] ; then
4747
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
4848
else
49-
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 -U
49+
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 -U
5050
fi
5151
elif [[ "$TORCH_VERSION" == "stable" ]]; then
5252
if [ "${CU_VERSION:-}" == cpu ] ; then
5353
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
5454
else
55-
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124
55+
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128
5656
fi
5757
else
5858
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_d4rl/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,18 +26,18 @@ fi
2626
# submodules
2727
git submodule sync && git submodule update --init --recursive
2828

29-
printf "Installing PyTorch with cu124"
29+
printf "Installing PyTorch with cu128"
3030
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3131
if [ "${CU_VERSION:-}" == cpu ] ; then
3232
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
3333
else
34-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
34+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3535
fi
3636
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3737
if [ "${CU_VERSION:-}" == cpu ] ; then
3838
pip3 install torch --index-url https://download.pytorch.org/whl/cpu
3939
else
40-
pip3 install torch --index-url https://download.pytorch.org/whl/cu124
40+
pip3 install torch --index-url https://download.pytorch.org/whl/cu128
4141
fi
4242
else
4343
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_envpool/install.sh‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,11 @@ fi
2626
# submodules
2727
git submodule sync && git submodule update --init --recursive
2828

29-
printf "Installing PyTorch with cu124"
29+
printf "Installing PyTorch with cu128"
3030
if [ "${CU_VERSION:-}" == cpu ] ; then
3131
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
3232
else
33-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
33+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3434
fi
3535

3636
# smoke test

‎.github/unittest/linux_libs/scripts_gen-dgrl/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,18 +26,18 @@ fi
2626
# submodules
2727
git submodule sync && git submodule update --init --recursive
2828

29-
printf "Installing PyTorch with cu124"
29+
printf "Installing PyTorch with cu128"
3030
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3131
if [ "${CU_VERSION:-}" == cpu ] ; then
3232
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
3333
else
34-
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 -U
34+
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3535
fi
3636
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3737
if [ "${CU_VERSION:-}" == cpu ] ; then
3838
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
3939
else
40-
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124
40+
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128
4141
fi
4242
else
4343
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_habitat/install.sh‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@ git submodule sync && git submodule update --init --recursive
2121

2222
printf "Installing PyTorch with %s\n" "${CU_VERSION}"
2323
if [[ "$TORCH_VERSION" == "nightly" ]]; then
24-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
24+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
2525
elif [[ "$TORCH_VERSION" == "stable" ]]; then
26-
pip3 install torch --index-url https://download.pytorch.org/whl/cu124
26+
pip3 install torch --index-url https://download.pytorch.org/whl/cu128
2727
fi
2828

2929
# install tensordict

‎.github/unittest/linux_libs/scripts_jumanji/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,18 +25,18 @@ fi
2525
# submodules
2626
git submodule sync && git submodule update --init --recursive
2727

28-
printf "Installing PyTorch with cu124"
28+
printf "Installing PyTorch with cu128"
2929
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3030
if [ "${CU_VERSION:-}" == cpu ] ; then
3131
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U
3232
else
33-
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 -U
33+
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3434
fi
3535
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3636
if [ "${CU_VERSION:-}" == cpu ] ; then
3737
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cpu
3838
else
39-
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124
39+
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128
4040
fi
4141
else
4242
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_meltingpot/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,18 +25,18 @@ fi
2525
# submodules
2626
git submodule sync && git submodule update --init --recursive
2727

28-
printf "Installing PyTorch with cu124"
28+
printf "Installing PyTorch with cu128"
2929
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3030
if [ "${CU_VERSION:-}" == cpu ] ; then
3131
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
3232
else
33-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
33+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3434
fi
3535
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3636
if [ "${CU_VERSION:-}" == cpu ] ; then
3737
pip3 install torch --index-url https://download.pytorch.org/whl/cpu
3838
else
39-
pip3 install torch --index-url https://download.pytorch.org/whl/cu124
39+
pip3 install torch --index-url https://download.pytorch.org/whl/cu128
4040
fi
4141
else
4242
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_minari/install.sh‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,18 +26,18 @@ fi
2626
# submodules
2727
git submodule sync && git submodule update --init --recursive
2828

29-
printf "Installing PyTorch with cu124"
29+
printf "Installing PyTorch with cu128"
3030
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3131
if [ "${CU_VERSION:-}" == cpu ] ; then
3232
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
3333
else
34-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
34+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
3535
fi
3636
elif [[ "$TORCH_VERSION" == "stable" ]]; then
3737
if [ "${CU_VERSION:-}" == cpu ] ; then
3838
pip3 install torch --index-url https://download.pytorch.org/whl/cpu
3939
else
40-
pip3 install torch --index-url https://download.pytorch.org/whl/cu124
40+
pip3 install torch --index-url https://download.pytorch.org/whl/cu128
4141
fi
4242
else
4343
printf "Failed to install pytorch"

0 commit comments

Comments
 (0)