GPU版本安装Pytorch教程最新指南及问题解决办法

383 阅读1分钟

下载安装CUDA11.3

CUDA是Nvidia推出的只能用于自家GPU的并行计算框架。只有安装这个框架才能够进行复杂的并行计算

cudnn是一个GPU 加速的深度神经网络基元库**,能够以高度优化的方式实现标准例程(如前向和反向卷积、池化层、归一化和激活层)

详细了解可参考大神的文章

blog.csdn.net/u014380165/…

查看对应版本的CUDA,这部非常关键!!!请一定要重视,避免之后多次重装。

(1)首先查看自己电脑GPU版本

win+R打开cmd,输入nvidia-smi

nvidia-smi.png

(2)查看自己对应的版本

docs.nvidia.com/cuda/cuda-t…

CUDA下载链接:developer.nvidia.com/cuda-toolki…

接下来,我们来下载这个CUDA,这里选择的版本不能高于你的显卡驱动里面那个版本号,由于我的是12.1,且系统是win10,所以我这里选择的是12.1

安装

53d55cda5e5b43abb3e2be775b00cd79.png

这里与其他安装方法不同的是,我没有安装 cudnn,也没有配置 path ,但是同样也安装成功。

打开 Anaconda prompt 命令,输入命令:

nvcc -V

下载GPU版本下的pytorch和pytorchvision

下载 pytorch 的下载链接为:

download.pytorch.org/whl/torch_s…

里面的文件是cpu开头的是 CPU 版本,cu开头的才是我们要下载的GPU版本。
这里可参考 torch 和 torchvision 的对应图,以免下错对应版本,这里我选择的是红色圈内的,根据自己需求下载

11.png

下载后放在同一目录下,我在d盘新建文件夹 CUDA-python,放在了E:\CUDA-pytorch下:

33.png

下载完成后切换到下载目录,使用pip命令安装。

检验,输入命令:

pip list 

55.png

验证

  1. import torch
  2. torch.cuda.is_available()
  3. True True代表安装成功