GoogleCloudPlatform / GoogleCloudPlatform/compute-gpu-installation
Make the CUDA version pytorch compatible
Open
type: feature request
- Dominant language
- Python
- Stars
- 89
- Forks
- 40
- PR merge metrics
- No merged PRs in 30d
Description
I'm on Ubuntu 22.04 ran this script to set up and it installed CUDA 11.5. Unfortunately pytorch only supports CUDA 10.2, 11.3, and 11.6. Could we make it the default, or at least an option, to install a version of CUDA that is supported by pytorch?
Contributor guide
Assessment
This issue has not been assessed yet.