Skip to content

Commit 7342b94

Browse files
ptrblckptrblck
committed
remove CUDA 11.6 builds
1 parent 669cbd4 commit 7342b94

File tree

4 files changed

+3
-92
lines changed

4 files changed

+3
-92
lines changed

‎CUDA_UPGRADE_GUIDE.MD

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@ Here is the supported matrix for CUDA and CUDNN
99

1010
| CUDA | CUDNN | additional details |
1111
| --- | --- | --- |
12-
| 11.6 | 8.3.2.44 | Stable CUDA Release |
13-
| 11.7 | 8.5.0.96 | Latest CUDA Release |
12+
| 11.7 | 8.5.0.96 | Stable CUDA Release |
13+
| 11.8 | 8.7.0.84 | Latest CUDA Release |
1414

1515

1616
### B. Check the package availability

‎common/install_cuda.sh

Lines changed: 0 additions & 54 deletions
Original file line numberDiff line numberDiff line change
@@ -2,27 +2,6 @@
22

33
set -ex
44

5-
function install_116 {
6-
echo "Installing CUDA 11.6 and CuDNN 8.3"
7-
rm -rf /usr/local/cuda-11.6 /usr/local/cuda
8-
# install CUDA 11.6.2 in the same container
9-
wget -q https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
10-
chmod +x cuda_11.6.2_510.47.03_linux.run
11-
./cuda_11.6.2_510.47.03_linux.run --toolkit --silent
12-
rm -f cuda_11.6.2_510.47.03_linux.run
13-
rm -f /usr/local/cuda && ln -s /usr/local/cuda-11.6 /usr/local/cuda
14-
15-
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
16-
mkdir tmp_cudnn && cd tmp_cudnn
17-
wget -q https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz -O cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
18-
tar xf cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
19-
cp -a cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/include/* /usr/local/cuda/include/
20-
cp -a cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/lib/* /usr/local/cuda/lib64/
21-
cd ..
22-
rm -rf tmp_cudnn
23-
ldconfig
24-
}
25-
265
function install_117 {
276
echo "Installing CUDA 11.7 and CuDNN 8.5 and NCCL 2.14"
287
rm -rf /usr/local/cuda-11.7 /usr/local/cuda
@@ -85,37 +64,6 @@ function install_118 {
8564
ldconfig
8665
}
8766

88-
function prune_116 {
89-
echo "Pruning CUDA 11.6 and CuDNN"
90-
#####################################################################################
91-
# CUDA 11.6 prune static libs
92-
#####################################################################################
93-
export NVPRUNE="/usr/local/cuda-11.6/bin/nvprune"
94-
export CUDA_LIB_DIR="/usr/local/cuda-11.6/lib64"
95-
96-
export GENCODE="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86"
97-
export GENCODE_CUDNN="-gencode arch=compute_35,code=sm_35 -gencode arch=compute_37,code=sm_37 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_75,code=sm_75 -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86"
98-
99-
if [[ -n "$OVERRIDE_GENCODE" ]]; then
100-
export GENCODE=$OVERRIDE_GENCODE
101-
fi
102-
103-
# all CUDA libs except CuDNN and CuBLAS (cudnn and cublas need arch 3.7 included)
104-
ls $CUDA_LIB_DIR/ | grep "\.a" | grep -v "culibos" | grep -v "cudart" | grep -v "cudnn" | grep -v "cublas" | grep -v "metis" \
105-
| xargs -I {} bash -c \
106-
"echo {} && $NVPRUNE $GENCODE $CUDA_LIB_DIR/{} -o $CUDA_LIB_DIR/{}"
107-
108-
# prune CuDNN and CuBLAS
109-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublas_static.a -o $CUDA_LIB_DIR/libcublas_static.a
110-
$NVPRUNE $GENCODE_CUDNN $CUDA_LIB_DIR/libcublasLt_static.a -o $CUDA_LIB_DIR/libcublasLt_static.a
111-
112-
#####################################################################################
113-
# CUDA 11.6 prune visual tools
114-
#####################################################################################
115-
export CUDA_BASE="/usr/local/cuda-11.6/"
116-
rm -rf $CUDA_BASE/libnvvp $CUDA_BASE/nsightee_plugins $CUDA_BASE/nsight-compute-2022.1.1 $CUDA_BASE/nsight-systems-2021.5.2
117-
}
118-
11967
function prune_117 {
12068
echo "Pruning CUDA 11.7 and CuDNN"
12169
#####################################################################################
@@ -182,8 +130,6 @@ function prune_118 {
182130
while test $# -gt 0
183131
do
184132
case "$1" in
185-
11.6) install_116; prune_116
186-
;;
187133
11.7) install_117; prune_117
188134
;;
189135
11.8) install_118; prune_118

‎conda/pytorch-nightly/build.sh

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -55,14 +55,7 @@ if [[ -n "$build_with_cuda" ]]; then
5555
TORCH_CUDA_ARCH_LIST="3.7+PTX;5.0"
5656
export USE_STATIC_CUDNN=1 # links cudnn statically (driven by tools/setup_helpers/cudnn.py)
5757

58-
if [[ $CUDA_VERSION == 11.6* ]]; then
59-
TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
60-
#for cuda 11.5 we use cudnn 8.3.2.44 https://docs.nvidia.com/deeplearning/cudnn/release-notes/rel_8.html
61-
#which does not have single static libcudnn_static.a deliverable to link with
62-
export USE_STATIC_CUDNN=0
63-
#for cuda 11.5 include all dynamic loading libraries
64-
DEPS_LIST=(/usr/local/cuda/lib64/libcudnn*.so.8 /usr/local/cuda-11.6/extras/CUPTI/lib64/libcupti.so.11.6)
65-
elif [[ $CUDA_VERSION == 11.7* ]]; then
58+
if [[ $CUDA_VERSION == 11.7* ]]; then
6659
TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
6760
#for cuda 11.7 we use cudnn 8.5
6861
#which does not have single static libcudnn_static.a deliverable to link with

‎windows/internal/cuda_install.bat

Lines changed: 0 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -19,40 +19,12 @@ set CUDNN_LIB_FOLDER="lib\x64"
1919
:: Skip all of this if we already have cuda installed
2020
if exist "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\bin\nvcc.exe" goto set_cuda_env_vars
2121

22-
if %CUDA_VER% EQU 116 goto cuda116
2322
if %CUDA_VER% EQU 117 goto cuda117
2423
if %CUDA_VER% EQU 118 goto cuda118
2524

2625
echo CUDA %CUDA_VERSION_STR% is not supported
2726
exit /b 1
2827

29-
:cuda116
30-
31-
set CUDA_INSTALL_EXE=cuda_11.6.0_511.23_windows.exe
32-
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
33-
curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
34-
if errorlevel 1 exit /b 1
35-
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
36-
set "ARGS=thrust_11.6 nvcc_11.6 cuobjdump_11.6 nvprune_11.6 nvprof_11.6 cupti_11.6 cublas_11.6 cublas_dev_11.6 cudart_11.6 cufft_11.6 cufft_dev_11.6 curand_11.6 curand_dev_11.6 cusolver_11.6 cusolver_dev_11.6 cusparse_11.6 cusparse_dev_11.6 npp_11.6 npp_dev_11.6 nvrtc_11.6 nvrtc_dev_11.6 nvml_dev_11.6"
37-
)
38-
39-
set CUDNN_FOLDER=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive
40-
set CUDNN_LIB_FOLDER="lib"
41-
set "CUDNN_INSTALL_ZIP=%CUDNN_FOLDER%.zip"
42-
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
43-
curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
44-
if errorlevel 1 exit /b 1
45-
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
46-
)
47-
48-
@REM Cuda 8.3+ required zlib to be installed on the path
49-
echo Installing ZLIB dlls
50-
curl -k -L "http://s3.amazonaws.com/ossci-windows/zlib123dllx64.zip" --output "%SRC_DIR%\temp_build\zlib123dllx64.zip"
51-
7z x "%SRC_DIR%\temp_build\zlib123dllx64.zip" -o"%SRC_DIR%\temp_build\zlib"
52-
xcopy /Y "%SRC_DIR%\temp_build\zlib\dll_x64\*.dll" "C:\Windows\System32"
53-
54-
goto cuda_common
55-
5628
:cuda117
5729

5830
set CUDA_INSTALL_EXE=cuda_11.7.0_516.01_windows.exe

0 commit comments

Comments
 (0)