检查机器GPU型号,下载对应的NVIDIA驱动,一般驱动是向下兼容的,所以下载最新的即可。
百度或通过鲁大师查看。
NVIDIA 驱动程序下载:
下载NVIDIA驱动对应的CUDA:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
安装CUDA对应的PyTorch。
遇到问题可以安装Visual Studio Code解决。
检查机器GPU型号,下载对应的NVIDIA驱动,一般驱动是向下兼容的,所以下载最新的即可。
百度或通过鲁大师查看。
NVIDIA 驱动程序下载:
下载NVIDIA驱动对应的CUDA:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
安装CUDA对应的PyTorch。
遇到问题可以安装Visual Studio Code解决。