Unverified Commit 03701343 authored by Andrey Talman's avatar Andrey Talman Committed by GitHub
Browse files

Add missing cuda 12.1 code, cleanup cu116 use cases (#7558)

parent 14553fb9
...@@ -260,15 +260,15 @@ setup_conda_cudatoolkit_constraint() { ...@@ -260,15 +260,15 @@ setup_conda_cudatoolkit_constraint() {
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
else else
case "$CU_VERSION" in case "$CU_VERSION" in
cu121)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=12.1 # [not osx]"
;;
cu118) cu118)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.8 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.8 # [not osx]"
;; ;;
cu117) cu117)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.7 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.7 # [not osx]"
;; ;;
cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]"
;;
cpu) cpu)
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
...@@ -289,15 +289,15 @@ setup_conda_cudatoolkit_plain_constraint() { ...@@ -289,15 +289,15 @@ setup_conda_cudatoolkit_plain_constraint() {
export CMAKE_USE_CUDA=0 export CMAKE_USE_CUDA=0
else else
case "$CU_VERSION" in case "$CU_VERSION" in
cu121)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=12.1"
;;
cu118) cu118)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.8" export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.8"
;; ;;
cu117) cu117)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.7" export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.7"
;; ;;
cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.6"
;;
cpu) cpu)
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
......
...@@ -23,7 +23,6 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR% ...@@ -23,7 +23,6 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
set CUDNN_FOLDER="cuda" set CUDNN_FOLDER="cuda"
set CUDNN_LIB_FOLDER="lib\x64" set CUDNN_LIB_FOLDER="lib\x64"
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 if %CUDA_VER% EQU 121 goto cuda121
......
...@@ -2,15 +2,5 @@ ...@@ -2,15 +2,5 @@
set -ex set -ex
if [[ "$CU_VERSION" == "cu92" ]]; then export VC_YEAR=2019
export VC_YEAR=2017 export VSDEVCMD_ARGS=""
export VSDEVCMD_ARGS="-vcvars_ver=14.13"
powershell packaging/windows/internal/vs2017_install.ps1
elif [[ "$CU_VERSION" == "cu100" ]]; then
export VC_YEAR=2017
export VSDEVCMD_ARGS=""
powershell packaging/windows/internal/vs2017_install.ps1
else
export VC_YEAR=2019
export VSDEVCMD_ARGS=""
fi
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