pytorch的cuda环境搭建(GPU版本安装)

2022-10-08 11:06:35

第一步:安装cuda

安装cuda,这里还需要安装对应版本的Visual Studio(具体参考本博主的博客https://blog.csdn.net/qq_36653505/article/details/81368346

第二步:安装cudnn

cudnn是pytorch搭建深度学习模型的依赖,没有它,不能运行卷积等操作。
下载对应版本的cudnn链接:https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述
打开压缩包,依次将压缩包中的3个文件复制到相应位置

  1. C:\cuda\bin\cudnn64_7.dll —> C:\Program Files\NVIDIA GPUComputing
    Toolkit\CUDA\v9.1\bin

  2. C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing
    Toolkit\CUDA\v9.1\include

  3. C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing
    Toolkit\CUDA\v9.1\lib\x64

注意: 一定要安装cudnn,不然无法进行卷积操作

安装GPU版本的pytorch

到官网下载对应GPU版本的pytorch
在这里插入图片描述

然后使用pip install 文件名安装下载好的pytorch压缩包

第四步:检查是否安装成功

在这里插入图片描述

参考

  1. https://blog.csdn.net/MarsLee_U/article/details/80489618
  2. https://blog.csdn.net/u011622208/article/details/80275344
  • 作者:Never-Giveup
  • 原文链接:https://blog.csdn.net/qq_36653505/article/details/83932941
    更新时间:2022-10-08 11:06:35