Skip to content

Commit bd9cb2d

Browse files
committed
Update NVSHMEM version in CUDA installation script to 3.3.24 and modify download URL structure
1 parent be4e041 commit bd9cb2d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

install_cuda.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ echo "🚀 ===== CUDA Installation Script Started ====="
1414
CUDA_INSTALL_PREFIX=${CUDA_INSTALL_PREFIX:-$HOME/opt}
1515
CUDA_INSTALL_PREFIX=${CUDA_INSTALL_PREFIX%/}
1616
CUDA_VERSION=${CUDA_VERSION:-12.8}
17-
NVSHMEM_VERSION=${NVSHMEM_VERSION:-3.3.20}
17+
NVSHMEM_VERSION=${NVSHMEM_VERSION:-3.3.24}
1818
INSTALL_NCCL=${INSTALL_NCCL:-1}
1919

2020
echo "CUDA_INSTALL_PREFIX=${CUDA_INSTALL_PREFIX}"
@@ -317,7 +317,7 @@ function install_nvshmem {
317317
# nvSHMEM license: https://docs.nvidia.com/nvshmem/api/sla.html
318318
local filename="libnvshmem-linux-${arch_path}-${nvshmem_version}_cuda${cuda_major_version}-archive"
319319
local suffix=".tar.xz"
320-
local url="https://developer.download.nvidia.com/compute/redist/nvshmem/${nvshmem_version}/builds/cuda${cuda_major_version}/txz/agnostic/${dl_arch}/${filename}${suffix}"
320+
local url="https://developer.download.nvidia.com/compute/nvshmem/redist/libnvshmem/linux-${arch_path}/${filename}${suffix}"
321321

322322
echo "Downloading nvSHMEM: ${filename}${suffix}"
323323
if ! wget -q "${url}"; then

0 commit comments

Comments
 (0)