diff --git a/ports/cuda/CONTROL b/ports/cuda/CONTROL index cf909e0cefb..00b422df859 100644 --- a/ports/cuda/CONTROL +++ b/ports/cuda/CONTROL @@ -1,4 +1,5 @@ Source: cuda -Version: 10.1-1 +Version: 10.1 +Port-Version: 2 Description: A parallel computing platform and programming model Homepage: https://developer.nvidia.com/cuda-toolkit diff --git a/ports/cuda/portfile.cmake b/ports/cuda/portfile.cmake index 63c761a1ae0..0dd768e29d2 100644 --- a/ports/cuda/portfile.cmake +++ b/ports/cuda/portfile.cmake @@ -5,12 +5,18 @@ #note: this port must be kept in sync with CUDNN port: every time one is upgraded, the other must be too set(CUDA_REQUIRED_VERSION "10.1.0") +set(CUDA_PATHS + ENV CUDA_PATH + ENV CUDA_BIN_PATH + ENV CUDA_PATH_V11_0 + ENV CUDA_PATH_V10_2 + ENV CUDA_PATH_V10_1) + if (VCPKG_TARGET_IS_WINDOWS) find_program(NVCC NAMES nvcc.exe PATHS - ENV CUDA_PATH - ENV CUDA_BIN_PATH + ${CUDA_PATHS} PATH_SUFFIXES bin bin64 DOC "Toolkit location." NO_DEFAULT_PATH @@ -38,8 +44,7 @@ else() find_program(NVCC NAMES nvcc PATHS - ENV CUDA_PATH - ENV CUDA_BIN_PATH + ${CUDA_PATHS} PATHS ${FOUND_PATH} PATH_SUFFIXES bin bin64 DOC "Toolkit location." @@ -57,6 +62,7 @@ endif() if (error_code) + message(STATUS "Executing ${NVCC} --version resulted in error: ${error_code}") message(FATAL_ERROR "Could not find CUDA. Before continuing, please download and install CUDA (v${CUDA_REQUIRED_VERSION} or higher) from:" "\n https://developer.nvidia.com/cuda-downloads\n") endif() diff --git a/toolsrc/src/vcpkg/base/system.process.cpp b/toolsrc/src/vcpkg/base/system.process.cpp index d709fdb607c..2feaa3d74ca 100644 --- a/toolsrc/src/vcpkg/base/system.process.cpp +++ b/toolsrc/src/vcpkg/base/system.process.cpp @@ -308,6 +308,8 @@ namespace vcpkg L"CUDA_PATH_V9_1", L"CUDA_PATH_V10_0", L"CUDA_PATH_V10_1", + L"CUDA_PATH_V10_2", + L"CUDA_PATH_V11_0", L"CUDA_TOOLKIT_ROOT_DIR", // Environmental variable generated automatically by CUDA after installation L"NVCUDASAMPLES_ROOT",