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/

posted @ 2024-09-09 20:03  CVE_2003_0102  阅读(12)  评论(0编辑  收藏  举报