cuda 运行python cuda9.1 pytorch

搞了两天查看了无数帖子终于搞定了！！！记录一下

如果前期环境都准备好了（python、cuda），只是遇到以下情况，可以直接看第二部分pytorch安装过程

>>>import torch  //导包
>>>print(torch.cuda.is_available())  //验证cuda是否可用
>>>False  //不能

目录

一、准备工作：

1. 查看GPU

方法一：设备管理器

方法二：任务管理器，打开方式可同上

2. 如何查看电脑是否有CUDA及其版本

CUDA 安装教程

二、安装pytorch

注意：我是将pytorch安装在虚拟环境中，因此记得切换到虚拟环境，使用完毕后退出

1. 踩坑记录

2. 卸载pytorch

3. pip手动下载whl文件安装pytorch

---

一、准备工作：

1. 查看GPU

方法一：设备管理器

 

方法二：任务管理器，打开方式可同上

2. 如何查看电脑是否有CUDA及其版本

#两种都可
nvcc -V
nvcc --version

如果查不到很可能是没有安装，我前期一直查不到，但是去NIVIDA控制面板看是有的

但网上说这里显示的9.2.217是电脑能支持的最高版本。

反正我通过各种方式查找我电脑应该都是安装了的，但当时一头雾水还是去重新安装了一下。

CUDA 安装教程

参考：【CUDA】cuda安装 （windows版）

二、安装pytorch

注意：我是将pytorch安装在虚拟环境中，因此记得切换到虚拟环境，使用完毕后退出

因为在看《动手学深度学习》这本书，前期是跟着书上安装的，使用的是d2l环境

退出虚拟环境

conda deactivate #退出虚拟环境

conda create --name d2l python=3.8 -y #创建虚拟环境
conda activate d2l #激活虚拟环境

 到安装pytorch的时候出了问题，搞了这么多天，烦死

1. 踩坑记录

官网pytorch官网

 官网安装对应cuda9.2版本的安装之后无法使用

>>>import torch  //导包
>>>print(torch.cuda.is_available())  //验证cuda是否可用
>>>False  //不能

 原因：这种原因可能是下载版本不对，但我这里不是，因为我下载了对应版本，但是查阅资料发现官网代码直接下载对应cuda9.2版本的pytorch没有Windows版本，安装后pytorch是cpu版本的。解决办法：改为使用pip手动下载whl文件安装gpu版本

2. 卸载pytorch

如果已经下了pytorch但是不能用需要卸载的话，卸载方法：

conda uninstall pytorch  //之前使用conda安装
pip uninstall pytorch    //之前使用pip安装

一般直接这样卸载就可以卸载掉，但是我之前试过一次手动下载whl文件安装，又使用命令安装，安装得有点混乱然后发现whl文件下错了，也可以单独卸载

###使用的命令 后面带的地址就是存放whl文件的地址
pip install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
###卸载
pip uninstall torch
pip uninstall torchvision
pip uninstall torchaudio

3. pip手动下载whl文件安装pytorch

下载地址：pytorch || torchvision

cu92: cuda9.2版本

cp38: python3.8版本

win_amd64: windows64x

以下是我下载的版本

 

 下载后在下载目录进入cmd，使用命令

pip install 文件名.whl

如果不是在其下载目录则使用

pip install 下载目录/文件名.whl

而且手动下载速度比直接pip安装快太多

下载后可以命令行查看

pip list

 也可以通过python输出其版本以及验证cuda是否可用

import torch        
print(torch.__version__)        #查看版本
print(torch.cuda.is_available())#是否可用

 

 输出True，说明cuda可以使用