Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Handle RuntimeError thrown by CUDA Python in
validate_setup
(#10653)
The call to `getDeviceCount()` can raise a `RuntimeError` when `libcuda.so` is missing. We should handle that too in `validate_setup()`. Authors: - Ashwin Srinath (https://github.com/shwina) Approvers: - Charles Blackmon-Luca (https://github.com/charlesbluca) - GALI PREM SAGAR (https://github.com/galipremsagar) URL: #10653
- Loading branch information