Skip to content

Commit 7897b9f

Browse files
committed
Update prepare_build_environment_linux.sh to rhel7
1 parent 271fbb8 commit 7897b9f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/tools/prepare_build_environment_linux.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ if [ "$CIBW_ARCHS" == "aarch64" ]; then
2020

2121
else
2222
# Install CUDA 12.2:
23-
yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
23+
yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
2424
# error mirrorlist.centos.org doesn't exists anymore.
2525
sed -i s/mirror.centos.org/vault.centos.org/g /etc/yum.repos.d/*.repo
2626
sed -i s/^#.*baseurl=http/baseurl=http/g /etc/yum.repos.d/*.repo
@@ -29,9 +29,9 @@ else
2929
cuda-nvcc-12-2-12.2.140-1 \
3030
cuda-cudart-devel-12-2-12.2.140-1 \
3131
libcurand-devel-12-2-10.3.3.141-1 \
32-
libcudnn9-devel-cuda-12-9.1.0.70-1 \
32+
libcudnn9-devel-cuda-12-9.0.0.312-1 \
3333
libcublas-devel-12-2-12.2.5.6-1 \
34-
libnccl-2.18.5-1+cuda12.2
34+
libnccl-devel-2.19.3-1+cuda12.2
3535
ln -s cuda-12.2 /usr/local/cuda
3636

3737
ONEAPI_VERSION=2023.2.0

0 commit comments

Comments
 (0)