Skip to content

Commit 8c9dc05

Browse files
vmoensvmoens
committed
[CI] Upgrade to cuda 12.8
ghstack-source-id: e0ad7d6c00d53b74b23022836535c453a37df238 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

+3-3
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

+3-3
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

+3-3
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

+3-3
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

+2-2
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

+3-3
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

+2-2
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

+3-3
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

+3-3
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

+3-3
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_open_spiel/install.sh

+3-3
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_openx/install.sh

+3-3
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 -U
3939
else
40-
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124 -U
40+
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128 -U
4141
fi
4242
else
4343
printf "Failed to install pytorch"

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

+3-3
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_rlhf/install.sh

+3-3
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 "numpy<2.0.0" --index-url https://download.pytorch.org/whl/nightly/cpu -U
3333
else
34-
pip3 install --pre torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/nightly/cu124 -U
34+
pip3 install --pre torch "numpy<2.0.0" --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 "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cpu
3939
else
40-
pip3 install torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cu124
40+
pip3 install torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cu128
4141
fi
4242
else
4343
printf "Failed to install pytorch"

‎.github/unittest/linux_libs/scripts_robohive/install_and_run_test.sh

+3-3
Original file line numberDiff line numberDiff line change
@@ -34,18 +34,18 @@ esac
3434
# submodules
3535
git submodule sync && git submodule update --init --recursive
3636

37-
printf "Installing PyTorch with cu124"
37+
printf "Installing PyTorch with cu128"
3838
if [[ "$TORCH_VERSION" == "nightly" ]]; then
3939
if [ "${CU_VERSION:-}" == cpu ] ; then
4040
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu -U
4141
else
42-
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu124 -U
42+
pip3 install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128 -U
4343
fi
4444
elif [[ "$TORCH_VERSION" == "stable" ]]; then
4545
if [ "${CU_VERSION:-}" == cpu ] ; then
4646
pip3 install torch --index-url https://download.pytorch.org/whl/cpu
4747
else
48-
pip3 install torch --index-url https://download.pytorch.org/whl/cu124
48+
pip3 install torch --index-url https://download.pytorch.org/whl/cu128
4949
fi
5050
else
5151
printf "Failed to install pytorch"

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

+3-3
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_sklearn/install.sh

+3-3
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_smacv2/install.sh

+3-3
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_unity_mlagents/install.sh

+3-3
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"

0 commit comments

Comments
 (0)