CUDA9.2+Python3.8+Windows 10安装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可以使用

  • 6
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值