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
......@@ -147,7 +147,7 @@ if "%CUDA_VERSION%" == "cpu" (
if "%TORCHVISION_BUILD_VERSION%" == "" set TORCHVISION_BUILD_VERSION=0.6.0.dev%NIGHTLIES_DATE_COMPACT%
if "%~1" == "Wheels" (
if not "%CUDA_VERSION%" == "101" (
if not "%CUDA_VERSION%" == "102" (
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