diff --git a/test_example_code/CMakeLists.txt b/test_example_code/CMakeLists.txt index 1724a6ed0..e17ecf821 100644 --- a/test_example_code/CMakeLists.txt +++ b/test_example_code/CMakeLists.txt @@ -10,7 +10,7 @@ target_include_directories(simple-torch-test PRIVATE ${TORCH_INCLUDE_DIRS}) target_link_libraries(simple-torch-test "${TORCH_LIBRARIES}") set_property(TARGET simple-torch-test PROPERTY CXX_STANDARD 17) -find_package(CUDAToolkit 11.8) +find_package(CUDAToolkit 12.1) target_link_libraries(simple-torch-test CUDA::cudart CUDA::cufft CUDA::cusparse CUDA::cublas CUDA::cusolver) find_library(CUDNN_LIBRARY NAMES cudnn) diff --git a/windows/internal/static_lib_test.bat b/windows/internal/static_lib_test.bat index aa15dd68f..ffd2f8406 100644 --- a/windows/internal/static_lib_test.bat +++ b/windows/internal/static_lib_test.bat @@ -111,7 +111,7 @@ call "%VS15VCVARSALL%" x64 set install_root=%CD% set INCLUDE=%INCLUDE%;%install_root%\include;%install_root%\include\torch\csrc\api\include set LIB=%LIB%;%install_root%\lib\x64 -set PATH=%PATH%;%install_root%\lib +set PATH=%PATH%;%install_root%\lib;%CONDA_HOME%\envs\testenv\Library\lib cd %BUILDER_ROOT%\test_example_code\