From c604ba45c9576af9b48ae62943d4bd42fb9137c5 Mon Sep 17 00:00:00 2001 From: Guangyun Han Date: Tue, 20 Jul 2021 11:28:20 +0800 Subject: [PATCH] package cupti library with whl --- windows/internal/copy.bat | 1 + 1 file changed, 1 insertion(+) diff --git a/windows/internal/copy.bat b/windows/internal/copy.bat index a2907f3a5..a96821b3f 100755 --- a/windows/internal/copy.bat +++ b/windows/internal/copy.bat @@ -7,6 +7,7 @@ copy "%CUDA_PATH%\bin\cusolver*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\bin\cudnn*64_*.dll*" pytorch\torch\lib copy "%CUDA_PATH%\bin\nvrtc*64_*.dll*" pytorch\torch\lib +copy "%CUDA_PATH%\extras\CUPTI\lib64\cupti64_*.dll*" pytorch\torch\lib copy "C:\Program Files\NVIDIA Corporation\NvToolsExt\bin\x64\nvToolsExt64_1.dll*" pytorch\torch\lib copy "%CONDA_LIB_PATH%\libiomp*5md.dll" pytorch\torch\lib