diff --git a/.github/workflows/cuda/cu102-Linux-env.sh b/.github/workflows/cuda/cu102-Linux-env.sh new file mode 100644 index 000000000..b672b197a --- /dev/null +++ b/.github/workflows/cuda/cu102-Linux-env.sh @@ -0,0 +1,9 @@ +#!/bin/bash + +CUDA_HOME=/usr/local/cuda-10.2 +LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} +PATH=${CUDA_HOME}/bin:${PATH} + +export FORCE_CUDA=1 +export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5" +export CUDA_HOME=/usr/local/cuda-10.2 \ No newline at end of file diff --git a/.github/workflows/cuda/cu102-Linux.sh b/.github/workflows/cuda/cu102-Linux.sh new file mode 100644 index 000000000..46fb05319 --- /dev/null +++ b/.github/workflows/cuda/cu102-Linux.sh @@ -0,0 +1,15 @@ +#!/bin/bash + +OS=ubuntu1804 + +wget -nv https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin +sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600 +wget -nv https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-${OS}-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb +sudo dpkg -i cuda-repo-${OS}-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb +sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub + +sudo apt-get -qq update +sudo apt install cuda cuda-nvcc-10-2 cuda-libraries-dev-10-2 +sudo apt clean + +rm -f https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-${OS}-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb \ No newline at end of file diff --git a/.github/workflows/cuda/cu113-Linux-env.sh b/.github/workflows/cuda/cu113-Linux-env.sh new file mode 100644 index 000000000..ae7720591 --- /dev/null +++ b/.github/workflows/cuda/cu113-Linux-env.sh @@ -0,0 +1,9 @@ +#!/bin/bash + +CUDA_HOME=/usr/local/cuda-11.3 +LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} +PATH=${CUDA_HOME}/bin:${PATH} + +export FORCE_CUDA=1 +export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" +export CUDA_HOME=/usr/local/cuda-11.3 \ No newline at end of file diff --git a/.github/workflows/cuda/cu113-Linux.sh b/.github/workflows/cuda/cu113-Linux.sh new file mode 100644 index 000000000..b89a7fb85 --- /dev/null +++ b/.github/workflows/cuda/cu113-Linux.sh @@ -0,0 +1,15 @@ +#!/bin/bash + +OS=ubuntu1804 + +wget -nv https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin +sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600 +wget -nv https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-${OS}-11-3-local_11.3.0-465.19.01-1_amd64.deb +sudo dpkg -i cuda-repo-${OS}-11-3-local_11.3.0-465.19.01-1_amd64.deb +sudo apt-key add /var/cuda-repo-${OS}-11-3-local/7fa2af80.pub + +sudo apt-get -qq update +sudo apt install cuda cuda-nvcc-11-3 cuda-libraries-dev-11-3 +sudo apt clean + +rm -f https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-${OS}-11-3-local_11.3.0-465.19.01-1_amd64.deb \ No newline at end of file diff --git a/.github/workflows/cuda/cu116-Linux-env.sh b/.github/workflows/cuda/cu116-Linux-env.sh new file mode 100644 index 000000000..61fe32597 --- /dev/null +++ b/.github/workflows/cuda/cu116-Linux-env.sh @@ -0,0 +1,9 @@ +#!/bin/bash + +CUDA_HOME=/usr/local/cuda-11.6 +LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} +PATH=${CUDA_HOME}/bin:${PATH} + +export FORCE_CUDA=1 +export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" +export CUDA_HOME=/usr/local/cuda-11.6 \ No newline at end of file diff --git a/.github/workflows/cuda/cu116-Linux.sh b/.github/workflows/cuda/cu116-Linux.sh new file mode 100644 index 000000000..e3e4e2af7 --- /dev/null +++ b/.github/workflows/cuda/cu116-Linux.sh @@ -0,0 +1,15 @@ +#!/bin/bash + +OS=ubuntu1804 + +wget -nv https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin +sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600 +wget -nv https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb +sudo dpkg -i cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb +sudo apt-key add /var/cuda-repo-${OS}-11-6-local/7fa2af80.pub + +sudo apt-get -qq update +sudo apt install cuda cuda-nvcc-11-6 cuda-libraries-dev-11-6 +sudo apt clean + +rm -f https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb \ No newline at end of file