mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Fix diagnostic message for CUDA version mismatch in cuda.cmake (#157370)
This PR fixes #157354 It fixes the issue in 'cmake/public/cuda.cmake' where a diagnostic message incorrectly showed an empty CUDA version when 'FindCUDA' and header-reported versions differed. The problem was caused by this line: set(${cuda_version_from_findcuda} ${CUDA_VERSION_STRING}) This incorrectly used the value of cuda_version_from_findcuda as a variable name. As a result the version string wasn't assigned and the error message omitted the version. This has been corrected to: set(cuda_version_from_findcuda ${CUDA_VERSION_STRING}) Now the diagnostic message properly displays the CUDA version reported by FindCUDA. Pull Request resolved: https://github.com/pytorch/pytorch/pull/157370 Approved by: https://github.com/soulitzer
This commit is contained in:
committed by
PyTorch MergeBot
parent
00ae620b9f
commit
4ff9b7fa31
@ -110,7 +110,7 @@ if(CUDA_FOUND)
|
||||
# Force CUDA to be processed for again next time
|
||||
# TODO: I'm not sure if this counts as an implementation detail of
|
||||
# FindCUDA
|
||||
set(${cuda_version_from_findcuda} ${CUDA_VERSION_STRING})
|
||||
set(cuda_version_from_findcuda ${CUDA_VERSION_STRING})
|
||||
unset(CUDA_TOOLKIT_ROOT_DIR_INTERNAL CACHE)
|
||||
# Not strictly necessary, but for good luck.
|
||||
unset(CUDA_VERSION CACHE)
|
||||
|
Reference in New Issue
Block a user