OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Use local FindCUDA for CMake < 3.7

Browse Source
This commit is contained in:
Orion Reblitz-Richardson 2018-03-22 21:27:17 -07:00
parent 3a84574c81
commit e881efde79
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmake/MiscCheck.cmake
View File

@ -260,6 +260,8 @@ if (MSVC)
# We know that VS2017 needs the new FindCUDA functionality, so we will
# simply enable it for the whole Windows build.
set(CAFFE2_CMAKE_USE_LOCAL_FINDCUDA ON)
elseif (${CMAKE_VERSION} VERSION_LESS 3.7 AND ${USE_CUDA})
set(CAFFE2_CMAKE_USE_LOCAL_FINDCUDA ON)
endif()
if (${CAFFE2_CMAKE_USE_LOCAL_FINDCUDA})