pytorch [GPU版] 安装教程

引言

众所周知啊，在我们安装 GPU 版的 pytorch 之前，需要安装 CUDA，CUDA 的安装教程在此不过多追叙，详情可以看官方文档。

版本一览

1. win 10
2. cuda：11.6
3. python：3.8
4. pytorch 三宝：
   

安装教程

1. 咱首先要做的是确认我们的 CUDA 的版本

   • 打开cmd，输入 nvidia-smi
     

2. 知道 CUDA 版本之后，去pytorch官网一键下载就好了

   • 复制下边的下载指令到 pycharm 命令行下载就行了
     
     

3. 查看 torch 以及 torchvision 的版本

   import torch
   import torchvision
   print("pytorch version:",torch.__version__)
   print("torchvision version:",torchvision.__version__)
   

   

注：这里如果你CUDA版本是11.6，并按照我上边的流程去弄，接下来会报一个错

如果 CUDA 是其它版本的可以忽略我下边说的直接点赞加收藏就好了【🐶】

报错详解

“NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend.”

这里解释一下为什么会报错，原因是你 CUDA11.6 在官方网首页直接下载的 pytorch 三宝合集包里边，torchvision 这个是默认给你下 cpu 版本的，所以这边咱需要换一下它的版本

1. 点击我下边指出的地方，这里是进去看 pytorch 历史版本

   

2. 找到你 CUDA 的版本，复制去下载就好了

   

3. 最后查看一下版本，发现 torchvision 变成了cu116，这才是对的GPU版本

   

创造亚当之兄弟抽支烟、点个赞