记一次pytorch安装

起因

记得自己之前安装并使用过pytorch，半年没用。今天再次使用出现了No module named ‘torch’。不知道为什么没有了，于是重新安装。记得上次安装挺麻烦的，所以记录一下，以便再次发生不知道怎么回事包没了的事。

安装过程

安装尝试

进入官网 https://pytorch.org/get-started/locally/

使用如下自动选好的配置 复制下面那句，放入了

在flowprint这个环境下执行（之前的旧环境）

因为有开了clash所以下载速度还不错？

出现问题

import torch可以运行
torch.cuda.is_available() 返回False

先查看cuda安装有没有问题：nvcc -V

(flowprint) C:\Users\vmuse>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243

没有问题

再看看gpu的情况
打开nvidia控制管理的时候显示“您未连接到nvidia gpu的显示器”
啊这。。。。？？？
使用360 选择启动项，将英伟达相关启动项打开，重启电脑后可以打开nvidia控制管理。

再次使用torch.cuda.is_available() 依然返回False

查看CUDNN
继续找问题 cudnn的版本为764 与4个版本的cuda可以对应

#if !defined(CUDNN_H_)
#define CUDNN_H_

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 4

在cmd中输入nvidia-smi,查看当前驱动的版本号，我的cuda版本是10.1

突然发现，刚刚下载时 下了10.2版本的cuda对应的pytorch
不知道是不是这个原因干脆就直接重新来个环境，进行安装pytorch
conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

成功！

torch.cuda.device_count() 查看GPU数量
torch.cuda.get_device_name(0) 查看使用的GPU名字