diff --git a/conda/pytorch-nightly/bld.bat b/conda/pytorch-nightly/bld.bat index 09dc1a816..8f9575f85 100644 --- a/conda/pytorch-nightly/bld.bat +++ b/conda/pytorch-nightly/bld.bat @@ -21,11 +21,6 @@ set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%desired_cuda% set CUDA_BIN_PATH=%CUDA_PATH%\bin set TORCH_NVCC_FLAGS=-Xfatbin -compress-all set TORCH_CUDA_ARCH_LIST=3.7+PTX;5.0 -if "%desired_cuda%" == "10.2" set TORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST%;6.0;6.1;7.0;7.5 -if "%desired_cuda%" == "11.3" ( - set TORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST%;6.0;6.1;7.0;7.5;8.0;8.6 - set TORCH_NVCC_FLAGS=-Xfatbin -compress-all --threads 2 -) if "%desired_cuda%" == "11.5" ( set TORCH_CUDA_ARCH_LIST=%TORCH_CUDA_ARCH_LIST%;6.0;6.1;7.0;7.5;8.0;8.6 set TORCH_NVCC_FLAGS=-Xfatbin -compress-all --threads 2 @@ -112,6 +107,8 @@ IF "%USE_SCCACHE%" == "1" ( if NOT "%build_with_cuda%" == "" ( copy "%CUDA_BIN_PATH%\cudnn*64_*.dll*" %SP_DIR%\torch\lib + copy "%CUDA_BIN_PATH%\cudart*64_*.dll*" %SP_DIR%\torch\lib + copy "%NVTOOLSEXT_PATH%\bin\x64\nvToolsExt64_*.dll*" %SP_DIR%\torch\lib :: cupti library file name changes aggressively, bundle it to avoid :: potential file name mismatch. copy "%CUDA_PATH%\extras\CUPTI\lib64\cupti64_*.dll*" %SP_DIR%\torch\lib