CMake: Prefer CUDAToolkit over CUDA if available

pytorch / tensorpipe

A tensor-aware point-to-point communication primitive for machine learning

Other

248 stars 77 forks source link

Open peterbell10 opened 2 years ago

peterbell10 commented 2 years ago

In newer CMake versions find_package(CUDA) is deprecated in favor of find_package(CUDATookit) and enable_language(CUDA).

This first tries to find CUDAToolkit and falls back to CUDA if it's not available.

cc @malfet