conda安装GPU版pytorch

问题

使用官网给的命令语句安装GPU版本pytorch，输入torch.cuda.is_available()来查询安装结果，出现False。经查询发现问题在conda官方源里没有cudatoolkit对应版本的pytorch，于是conda自动安装了cpu版本的pytorch。

conda list #查询安装的package

可以看到安装的都是cpu版本的

解决方案

该方案参考文章：https://www.pudn.com/news/625d6d45be9ad24cfa806af3.html
1.创建新环境

conda create -n gpu python=3.7

2.激活环境

activate gpu

3.安装cudatoolkit

conda install cudatoolkit=10.1 #对应计算机gpu版本

4.安装cudnn，如果不指定版本，在安装CUDA之后，会自动匹配对应版本的cudnn安装

conda install cudnn

5.官网下载torch torchvision torchaudio 相应版本轮子

https://download.pytorch.org/whl/cu101

6.转到轮子保存的本地目录，先安装torch，后安装另外两个

pip install 完整轮子文件名（包括文件类型）

PS：没有合适博主电脑版本的torchaudio，先不下载了
7.输入测试代码，成功