mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:07:10 +08:00
87d6831b2e
Add the almalinux docker for building magma-cuda 13.0 https://github.com/pytorch/pytorch/issues/159779 Also fixed the NVSHMEM download link Pull Request resolved: https://github.com/pytorch/pytorch/pull/160201 Approved by: https://github.com/atalman Co-authored-by: Andrey Talman <atalman@fb.com>
29 lines
817 B
Bash
29 lines
817 B
Bash
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
NCCL_VERSION=""
|
|
if [[ ${CUDA_VERSION:0:2} == "11" ]]; then
|
|
NCCL_VERSION=$(cat ci_commit_pins/nccl-cu11.txt)
|
|
elif [[ ${CUDA_VERSION:0:2} == "12" ]]; then
|
|
NCCL_VERSION=$(cat ci_commit_pins/nccl-cu12.txt)
|
|
elif [[ ${CUDA_VERSION:0:2} == "13" ]]; then
|
|
NCCL_VERSION=$(cat ci_commit_pins/nccl-cu13.txt)
|
|
else
|
|
echo "Unexpected CUDA_VERSION ${CUDA_VERSION}"
|
|
exit 1
|
|
fi
|
|
|
|
if [[ -n "${NCCL_VERSION}" ]]; then
|
|
# NCCL license: https://docs.nvidia.com/deeplearning/nccl/#licenses
|
|
# Follow build: https://github.com/NVIDIA/nccl/tree/master?tab=readme-ov-file#build
|
|
git clone -b $NCCL_VERSION --depth 1 https://github.com/NVIDIA/nccl.git
|
|
pushd nccl
|
|
make -j src.build
|
|
cp -a build/include/* /usr/local/cuda/include/
|
|
cp -a build/lib/* /usr/local/cuda/lib64/
|
|
popd
|
|
rm -rf nccl
|
|
ldconfig
|
|
fi
|