File tree 3 files changed +8
-19
lines changed
3 files changed +8
-19
lines changed Original file line number Diff line number Diff line change @@ -260,15 +260,15 @@ setup_conda_cudatoolkit_constraint() {
260
260
export CONDA_BUILD_VARIANT=" cpu"
261
261
else
262
262
case " $CU_VERSION " in
263
+ cu121)
264
+ export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda=12.1 # [not osx]"
265
+ ;;
263
266
cu118)
264
267
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda=11.8 # [not osx]"
265
268
;;
266
269
cu117)
267
270
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda=11.7 # [not osx]"
268
271
;;
269
- cu116)
270
- export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda=11.6 # [not osx]"
271
- ;;
272
272
cpu)
273
273
export CONDA_CUDATOOLKIT_CONSTRAINT=" "
274
274
export CONDA_BUILD_VARIANT=" cpu"
@@ -289,15 +289,15 @@ setup_conda_cudatoolkit_plain_constraint() {
289
289
export CMAKE_USE_CUDA=0
290
290
else
291
291
case " $CU_VERSION " in
292
+ cu121)
293
+ export CONDA_CUDATOOLKIT_CONSTRAINT=" pytorch-cuda=12.1"
294
+ ;;
292
295
cu118)
293
296
export CONDA_CUDATOOLKIT_CONSTRAINT=" pytorch-cuda=11.8"
294
297
;;
295
298
cu117)
296
299
export CONDA_CUDATOOLKIT_CONSTRAINT=" pytorch-cuda=11.7"
297
300
;;
298
- cu116)
299
- export CONDA_CUDATOOLKIT_CONSTRAINT=" pytorch-cuda=11.6"
300
- ;;
301
301
cpu)
302
302
export CONDA_CUDATOOLKIT_CONSTRAINT=" "
303
303
export CONDA_BUILD_VARIANT=" cpu"
Original file line number Diff line number Diff line change @@ -23,7 +23,6 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
23
23
set CUDNN_FOLDER = " cuda"
24
24
set CUDNN_LIB_FOLDER = " lib\x64"
25
25
26
- if %CUDA_VER% EQU 116 goto cuda116
27
26
if %CUDA_VER% EQU 117 goto cuda117
28
27
if %CUDA_VER% EQU 118 goto cuda118
29
28
if %CUDA_VER% EQU 121 goto cuda121
Original file line number Diff line number Diff line change 2
2
3
3
set -ex
4
4
5
- if [[ " $CU_VERSION " == " cu92" ]]; then
6
- export VC_YEAR=2017
7
- export VSDEVCMD_ARGS=" -vcvars_ver=14.13"
8
- powershell packaging/windows/internal/vs2017_install.ps1
9
- elif [[ " $CU_VERSION " == " cu100" ]]; then
10
- export VC_YEAR=2017
11
- export VSDEVCMD_ARGS=" "
12
- powershell packaging/windows/internal/vs2017_install.ps1
13
- else
14
- export VC_YEAR=2019
15
- export VSDEVCMD_ARGS=" "
16
- fi
5
+ export VC_YEAR=2019
6
+ export VSDEVCMD_ARGS=" "
You can’t perform that action at this time.
0 commit comments