Skip to content

GitLab

Unverified Commit 0d75d9e5 authored by ptrblck's avatar ptrblck Committed by GitHub
Browse files

Add CUDA 12.1 builds (#7533) 

Windows CUDA-12.1 failures are expected
parent 2925df7c
Show whitespace changes
Inline Side-by-side
Showing with 255 additions and 25 deletions
.circleci/config.yml
View file @ 0d75d9e5
...@@ -598,6 +598,15 @@ workflows: ...@@ -598,6 +598,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.8_cu118 name: binary_win_wheel_py3.8_cu118
python_version: '3.8' python_version: '3.8'
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.8_cu121
python_version: '3.8'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -625,6 +634,15 @@ workflows: ...@@ -625,6 +634,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.9_cu118 name: binary_win_wheel_py3.9_cu118
python_version: '3.9' python_version: '3.9'
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.9_cu121
python_version: '3.9'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -652,6 +670,15 @@ workflows: ...@@ -652,6 +670,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.10_cu118 name: binary_win_wheel_py3.10_cu118
python_version: '3.10' python_version: '3.10'
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.10_cu121
python_version: '3.10'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
name: binary_win_wheel_py3.11_cpu name: binary_win_wheel_py3.11_cpu
...@@ -667,8 +694,17 @@ workflows: ...@@ -667,8 +694,17 @@ workflows:
python_version: '3.11' python_version: '3.11'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu118 cu_version: cu118
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.11_cu118 name: binary_win_wheel_py3.11_cu118
python_version: '3.11' python_version: '3.11'
- binary_win_wheel:
cu_version: cu121
name: binary_win_wheel_py3.11_cu121
python_version: '3.11'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -696,6 +732,15 @@ workflows: ...@@ -696,6 +732,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.8_cu118 name: binary_win_conda_py3.8_cu118
python_version: '3.8' python_version: '3.8'
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.8_cu121
python_version: '3.8'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -723,6 +768,15 @@ workflows: ...@@ -723,6 +768,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.9_cu118 name: binary_win_conda_py3.9_cu118
python_version: '3.9' python_version: '3.9'
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.9_cu121
python_version: '3.9'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -750,6 +804,15 @@ workflows: ...@@ -750,6 +804,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.10_cu118 name: binary_win_conda_py3.10_cu118
python_version: '3.10' python_version: '3.10'
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.10_cu121
python_version: '3.10'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
name: binary_win_conda_py3.11_cpu name: binary_win_conda_py3.11_cpu
...@@ -765,8 +828,17 @@ workflows: ...@@ -765,8 +828,17 @@ workflows:
python_version: '3.11' python_version: '3.11'
- binary_win_conda: - binary_win_conda:
cu_version: cu118 cu_version: cu118
filters:
branches:
only: main
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.11_cu118 name: binary_win_conda_py3.11_cu118
python_version: '3.11' python_version: '3.11'
- binary_win_conda:
cu_version: cu121
name: binary_win_conda_py3.11_cu121
python_version: '3.11'
- binary_ios_build: - binary_ios_build:
build_environment: binary-libtorchvision_ops-ios-12.0.0-x86_64 build_environment: binary-libtorchvision_ops-ios-12.0.0-x86_64
ios_arch: x86_64 ios_arch: x86_64
...@@ -928,6 +1000,26 @@ workflows: ...@@ -928,6 +1000,26 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.8_cu118 - nightly_binary_win_wheel_py3.8_cu118
subfolder: cu118/ subfolder: cu118/
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu121
python_version: '3.8'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu121_upload
requires:
- nightly_binary_win_wheel_py3.8_cu121
subfolder: cu121/
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -988,6 +1080,26 @@ workflows: ...@@ -988,6 +1080,26 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.9_cu118 - nightly_binary_win_wheel_py3.9_cu118
subfolder: cu118/ subfolder: cu118/
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.9_cu121
python_version: '3.9'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.9_cu121_upload
requires:
- nightly_binary_win_wheel_py3.9_cu121
subfolder: cu121/
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1048,6 +1160,26 @@ workflows: ...@@ -1048,6 +1160,26 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.10_cu118 - nightly_binary_win_wheel_py3.10_cu118
subfolder: cu118/ subfolder: cu118/
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.10_cu121
python_version: '3.10'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.10_cu121_upload
requires:
- nightly_binary_win_wheel_py3.10_cu121
subfolder: cu121/
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1108,6 +1240,26 @@ workflows: ...@@ -1108,6 +1240,26 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.11_cu118 - nightly_binary_win_wheel_py3.11_cu118
subfolder: cu118/ subfolder: cu118/
- binary_win_wheel:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.11_cu121
python_version: '3.11'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.11_cu121_upload
requires:
- nightly_binary_win_wheel_py3.11_cu121
subfolder: cu121/
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1165,6 +1317,25 @@ workflows: ...@@ -1165,6 +1317,25 @@ workflows:
name: nightly_binary_win_conda_py3.8_cu118_upload name: nightly_binary_win_conda_py3.8_cu118_upload
requires: requires:
- nightly_binary_win_conda_py3.8_cu118 - nightly_binary_win_conda_py3.8_cu118
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu121
python_version: '3.8'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu121_upload
requires:
- nightly_binary_win_conda_py3.8_cu121
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1222,6 +1393,25 @@ workflows: ...@@ -1222,6 +1393,25 @@ workflows:
name: nightly_binary_win_conda_py3.9_cu118_upload name: nightly_binary_win_conda_py3.9_cu118_upload
requires: requires:
- nightly_binary_win_conda_py3.9_cu118 - nightly_binary_win_conda_py3.9_cu118
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.9_cu121
python_version: '3.9'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.9_cu121_upload
requires:
- nightly_binary_win_conda_py3.9_cu121
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1279,6 +1469,25 @@ workflows: ...@@ -1279,6 +1469,25 @@ workflows:
name: nightly_binary_win_conda_py3.10_cu118_upload name: nightly_binary_win_conda_py3.10_cu118_upload
requires: requires:
- nightly_binary_win_conda_py3.10_cu118 - nightly_binary_win_conda_py3.10_cu118
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.10_cu121
python_version: '3.10'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.10_cu121_upload
requires:
- nightly_binary_win_conda_py3.10_cu121
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1336,6 +1545,25 @@ workflows: ...@@ -1336,6 +1545,25 @@ workflows:
name: nightly_binary_win_conda_py3.11_cu118_upload name: nightly_binary_win_conda_py3.11_cu118_upload
requires: requires:
- nightly_binary_win_conda_py3.11_cu118 - nightly_binary_win_conda_py3.11_cu118
- binary_win_conda:
cu_version: cu121
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.11_cu121
python_version: '3.11'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.11_cu121_upload
requires:
- nightly_binary_win_conda_py3.11_cu121
docker_build: docker_build:
triggers: triggers:
- schedule: - schedule:
......
.circleci/regenerate.py
View file @ 0d75d9e5
...@@ -32,8 +32,8 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6, ...@@ -32,8 +32,8 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
for os_type in ["linux", "macos", "win"]: for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS python_versions = PYTHON_VERSIONS
cu_versions_dict = { cu_versions_dict = {
"linux": ["cpu", "cu117", "cu118", "rocm5.2", "rocm5.3"], "linux": ["cpu", "cu117", "cu118", "cu121", "rocm5.2", "rocm5.3"],
"win": ["cpu", "cu117", "cu118"], "win": ["cpu", "cu117", "cu118", "cu121"],
"macos": ["cpu"], "macos": ["cpu"],
} }
cu_versions = cu_versions_dict[os_type] cu_versions = cu_versions_dict[os_type]
...@@ -109,6 +109,7 @@ def workflow_pair(btype, os_type, python_version, cu_version, unicode, prefix="" ...@@ -109,6 +109,7 @@ def workflow_pair(btype, os_type, python_version, cu_version, unicode, prefix=""
manylinux_images = { manylinux_images = {
"cu117": "pytorch/manylinux-cuda117", "cu117": "pytorch/manylinux-cuda117",
"cu118": "pytorch/manylinux-cuda118", "cu118": "pytorch/manylinux-cuda118",
"cu121": "pytorch/manylinux-cuda121",
} }
......
.circleci/unittest/windows/scripts/install.sh
View file @ 0d75d9e5
...@@ -25,7 +25,7 @@ else ...@@ -25,7 +25,7 @@ else
fi fi
cuda_toolkit_pckg="cudatoolkit" cuda_toolkit_pckg="cudatoolkit"
if [[ $CUDA_VERSION == 11.6 || $CUDA_VERSION == 11.7 || $CUDA_VERSION == 11.8 ]]; then if [[ $CUDA_VERSION == 11.6 || $CUDA_VERSION == 11.7 || $CUDA_VERSION == 11.8 || $CUDA_VERSION == 12.1 ]]; then
cuda_toolkit_pckg="pytorch-cuda" cuda_toolkit_pckg="pytorch-cuda"
fi fi
......
packaging/pkg_helpers.bash
View file @ 0d75d9e5
...@@ -46,13 +46,21 @@ setup_cuda() { ...@@ -46,13 +46,21 @@ setup_cuda() {
# Now work out the CUDA settings # Now work out the CUDA settings
case "$CU_VERSION" in case "$CU_VERSION" in
cu121)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.1"
else
export CUDA_HOME=/usr/local/cuda-12.1/
fi
export TORCH_CUDA_ARCH_LIST="5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
;;
cu118) cu118)
if [[ "$OSTYPE" == "msys" ]]; then if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.8" export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.8"
else else
export CUDA_HOME=/usr/local/cuda-11.8/ export CUDA_HOME=/usr/local/cuda-11.8/
fi fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
;; ;;
cu117) cu117)
if [[ "$OSTYPE" == "msys" ]]; then if [[ "$OSTYPE" == "msys" ]]; then
...@@ -62,14 +70,6 @@ setup_cuda() { ...@@ -62,14 +70,6 @@ setup_cuda() {
fi fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;; ;;
cu116)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.6"
else
export CUDA_HOME=/usr/local/cuda-11.6/
fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;;
cpu) cpu)
;; ;;
rocm*) rocm*)
......
packaging/windows/internal/cuda_install.bat
View file @ 0d75d9e5
...@@ -26,18 +26,19 @@ set CUDNN_LIB_FOLDER="lib\x64" ...@@ -26,18 +26,19 @@ set CUDNN_LIB_FOLDER="lib\x64"
if %CUDA_VER% EQU 116 goto cuda116 if %CUDA_VER% EQU 116 goto cuda116
if %CUDA_VER% EQU 117 goto cuda117 if %CUDA_VER% EQU 117 goto cuda117
if %CUDA_VER% EQU 118 goto cuda118 if %CUDA_VER% EQU 118 goto cuda118
if %CUDA_VER% EQU 121 goto cuda121
echo CUDA %CUDA_VERSION_STR% is not supported echo CUDA %CUDA_VERSION_STR% is not supported
exit /b 1 exit /b 1
:cuda116 :cuda117
set CUDA_INSTALL_EXE=cuda_11.6.0_511.23_windows.exe set CUDA_INSTALL_EXE=cuda_11.7.0_516.01_windows.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" ( if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
if errorlevel 1 exit /b 1 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
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 nvjpeg_11.6 nvjpeg_dev_11.6 nvrtc_11.6 nvrtc_dev_11.6 nvml_dev_11.6" set "ARGS=thrust_11.7 nvcc_11.7 cuobjdump_11.7 nvprune_11.7 nvprof_11.7 cupti_11.7 cublas_11.7 cublas_dev_11.7 cudart_11.7 cufft_11.7 cufft_dev_11.7 curand_11.7 curand_dev_11.7 cusolver_11.7 cusolver_dev_11.7 cusparse_11.7 cusparse_dev_11.7 npp_11.7 npp_dev_11.7 nvjpeg_11.7 nvjpeg_dev_11.7 nvrtc_11.7 nvrtc_dev_11.7 nvml_dev_11.7"
) )
set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip
...@@ -56,18 +57,18 @@ if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" ( ...@@ -56,18 +57,18 @@ if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
goto cuda_common goto cuda_common
:cuda117 :cuda118
set CUDA_INSTALL_EXE=cuda_11.7.0_516.01_windows.exe set CUDA_INSTALL_EXE=cuda_11.8.0_522.06_windows.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" ( if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
if errorlevel 1 exit /b 1 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
set "ARGS=thrust_11.7 nvcc_11.7 cuobjdump_11.7 nvprune_11.7 nvprof_11.7 cupti_11.7 cublas_11.7 cublas_dev_11.7 cudart_11.7 cufft_11.7 cufft_dev_11.7 curand_11.7 curand_dev_11.7 cusolver_11.7 cusolver_dev_11.7 cusparse_11.7 cusparse_dev_11.7 npp_11.7 npp_dev_11.7 nvjpeg_11.7 nvjpeg_dev_11.7 nvrtc_11.7 nvrtc_dev_11.7 nvml_dev_11.7" set "ARGS=cuda_profiler_api_11.8 thrust_11.8 nvcc_11.8 cuobjdump_11.8 nvprune_11.8 nvprof_11.8 cupti_11.8 cublas_11.8 cublas_dev_11.8 cudart_11.8 cufft_11.8 cufft_dev_11.8 curand_11.8 curand_dev_11.8 cusolver_11.8 cusolver_dev_11.8 cusparse_11.8 cusparse_dev_11.8 npp_11.8 npp_dev_11.8 nvjpeg_11.8 nvjpeg_dev_11.8 nvrtc_11.8 nvrtc_dev_11.8 nvml_dev_11.8"
) )
set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.5.0.96_cuda11-archive.zip
set CUDNN_FOLDER=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive set CUDNN_FOLDER=cudnn-windows-x86_64-8.5.0.96_cuda11-archive
set CUDNN_LIB_FOLDER="lib" set CUDNN_LIB_FOLDER="lib"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" ( if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
...@@ -82,18 +83,18 @@ if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" ( ...@@ -82,18 +83,18 @@ if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
goto cuda_common goto cuda_common
:cuda118 :cuda121
set CUDA_INSTALL_EXE=cuda_11.8.0_522.06_windows.exe set CUDA_INSTALL_EXE=cuda_12.1.0_531.14_windows.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" ( if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
if errorlevel 1 exit /b 1 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
set "ARGS=cuda_profiler_api_11.8 thrust_11.8 nvcc_11.8 cuobjdump_11.8 nvprune_11.8 nvprof_11.8 cupti_11.8 cublas_11.8 cublas_dev_11.8 cudart_11.8 cufft_11.8 cufft_dev_11.8 curand_11.8 curand_dev_11.8 cusolver_11.8 cusolver_dev_11.8 cusparse_11.8 cusparse_dev_11.8 npp_11.8 npp_dev_11.8 nvjpeg_11.8 nvjpeg_dev_11.8 nvrtc_11.8 nvrtc_dev_11.8 nvml_dev_11.8" set "ARGS=cuda_profiler_api_12.1 thrust_12.1 nvcc_12.1 cuobjdump_12.1 nvprune_12.1 nvprof_12.1 cupti_12.1 cublas_12.1 cublas_dev_12.1 cudart_12.1 cufft_12.1 cufft_dev_12.1 curand_12.1 curand_dev_12.1 cusolver_12.1 cusolver_dev_12.1 cusparse_12.1 cusparse_dev_12.1 npp_12.1 npp_dev_12.1 nvrtc_12.1 nvrtc_dev_12.1 nvml_dev_12.1"
) )
set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.5.0.96_cuda11-archive.zip set CUDNN_FOLDER=cudnn-windows-x86_64-8.8.1.3_cuda12-archive
set CUDNN_FOLDER=cudnn-windows-x86_64-8.5.0.96_cuda11-archive set "CUDNN_INSTALL_ZIP=%CUDNN_FOLDER%.zip"
set CUDNN_LIB_FOLDER="lib" set CUDNN_LIB_FOLDER="lib"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" ( if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment