Skip to content

Commit

Permalink
Fix for including nvtx dll and cudart (pytorch#1156)
Browse files Browse the repository at this point in the history
* Fix for invluding nvtx dll and cudart

* Fix for include nvtx

* Fix spaces
  • Loading branch information
atalman authored and jithunnair-amd committed Nov 1, 2022
1 parent 75c2ae6 commit f10a75c
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions conda/pytorch-nightly/bld.bat
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit f10a75c

Please sign in to comment.