安装GPU版本的pytorch

前言：最近新建了一个虚拟环境，但是在跑代码的时候出现问题“libc10_cuda.so: cannot open shared object file: No such file or directory”，去网上搜了一下，说是安装的pytorch为cpu版本的。网上给出的解决方案基本就两种，一种是直接卸载现有的pytorch版本，然后去官网找到对应的版本进行下载（但是官网给的命令下载速度太慢了），另外一种是添加镜像源然后进行下载但是！！！下载下来的仍然是cpu版本的！！

直到我在一篇文章（(14条消息) 安装GPU版本的pytorch（解决pytorch安装时默认安装CPU版本的问题）保姆级教程_3EoIPrime的博客-CSDN博客）的评论区看到一老哥评论，说直接用pip下载，才解决问题！

一、查看pytorch的版本

首先激活虚拟环境，然后输入conda list，在里面找到pytoch，可以看到版本号（图里面就是cpu的）

​

 二、查看python、cuda版本

输入：

python --version  #回车即可看到python版本

输入：

nvidia-smi #回车即可看到cuda版本 

​

 三、在官网（PyTorch）找到对应安装命令

 复制命令进行安装即可

四、检查是否可用

进入python

然后输入：import torch

没有报错接着输入：torch.cuda.is_available()

返回为true则没有问题

 同时可用查看torch版本，带了+cu的就OK

输入：print(torch.__version__)

五、补充

在安装好以后继续运行代码出现了一个问题

RuntimeError: nms_impl: implementation for device cuda:0 not found.

解决方法：卸载mmcv-full ，再重新安装  pip install mmcv-full

再次运行即可