Skip to content

GitLab

Unverified Commit 4f162503 authored by peterjc123's avatar peterjc123 Committed by GitHub
Browse files

Mark cu102 as the default for wheels on Windows (#2099)

parent abc15063
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
packaging/windows/internal/nightly_defaults.bat
View file @ 4f162503
...@@ -147,7 +147,7 @@ if "%CUDA_VERSION%" == "cpu" ( ...@@ -147,7 +147,7 @@ if "%CUDA_VERSION%" == "cpu" (
if "%TORCHVISION_BUILD_VERSION%" == "" set TORCHVISION_BUILD_VERSION=0.6.0.dev%NIGHTLIES_DATE_COMPACT% if "%TORCHVISION_BUILD_VERSION%" == "" set TORCHVISION_BUILD_VERSION=0.6.0.dev%NIGHTLIES_DATE_COMPACT%
if "%~1" == "Wheels" ( if "%~1" == "Wheels" (
if not "%CUDA_VERSION%" == "101" ( if not "%CUDA_VERSION%" == "102" (
set TORCHVISION_BUILD_VERSION=%TORCHVISION_BUILD_VERSION%+%_DESIRED_CUDA% set TORCHVISION_BUILD_VERSION=%TORCHVISION_BUILD_VERSION%+%_DESIRED_CUDA%
) )
) )
......
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