Skip to content

Commit c87da29

Browse files
authored
Merge branch 'main' into resize-use-interp-uint8
2 parents 6a271ec + 0370134 commit c87da29

File tree

3 files changed

+8
-19
lines changed

3 files changed

+8
-19
lines changed

packaging/pkg_helpers.bash

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -260,15 +260,15 @@ setup_conda_cudatoolkit_constraint() {
260260
export CONDA_BUILD_VARIANT="cpu"
261261
else
262262
case "$CU_VERSION" in
263+
cu121)
264+
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=12.1 # [not osx]"
265+
;;
263266
cu118)
264267
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.8 # [not osx]"
265268
;;
266269
cu117)
267270
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.7 # [not osx]"
268271
;;
269-
cu116)
270-
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]"
271-
;;
272272
cpu)
273273
export CONDA_CUDATOOLKIT_CONSTRAINT=""
274274
export CONDA_BUILD_VARIANT="cpu"
@@ -289,15 +289,15 @@ setup_conda_cudatoolkit_plain_constraint() {
289289
export CMAKE_USE_CUDA=0
290290
else
291291
case "$CU_VERSION" in
292+
cu121)
293+
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=12.1"
294+
;;
292295
cu118)
293296
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.8"
294297
;;
295298
cu117)
296299
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.7"
297300
;;
298-
cu116)
299-
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.6"
300-
;;
301301
cpu)
302302
export CONDA_CUDATOOLKIT_CONSTRAINT=""
303303
export CONDA_BUILD_VARIANT="cpu"

packaging/windows/internal/cuda_install.bat

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
2323
set CUDNN_FOLDER="cuda"
2424
set CUDNN_LIB_FOLDER="lib\x64"
2525

26-
if %CUDA_VER% EQU 116 goto cuda116
2726
if %CUDA_VER% EQU 117 goto cuda117
2827
if %CUDA_VER% EQU 118 goto cuda118
2928
if %CUDA_VER% EQU 121 goto cuda121

packaging/windows/internal/vc_install_helper.sh

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,5 @@
22

33
set -ex
44

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=""

0 commit comments

Comments
 (0)