2.2.2 PyTorch 2.0 GPU NVIDIA运行库的安装 ——CUDA+cuDNN安装教程
参考文章:
https://blog.csdn.net/mbdong/article/details/121769951
CUDA download:
https://developer.nvidia.com/cuda-12-2-0-download-archive
CUDNN download:
https://developer.nvidia.com/rdp/cudnn-archive
下载的cuDNN是一个压缩文件,将它解压并把所有的目录复制到CUDA安装主目录中(直接覆盖原来的目录),
CUDA存储最终路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
最后完成PyTorch 2.0.1 GPU版本的安装
https://pytorch.org/get-started/locally/