window安装torch环境--不踩坑须知!

window安装torch环境–不踩坑须知!

1. 查看电脑安装的cudn版本

进入cmd输入:

nvidia-smi

在这里插入图片描述

2.查找cuda对应的torch版本

  1. 在官网查找:https://pytorch.org/get-started/previous-versions/
    在这里插入图片描述
  2. 如果呢你在新建的conda环境里输入该命令,百分之99无法安装完全,具体不知道原因,这里conda新建环境不用我说了吧,不懂的自行查吧,攻略一大堆,命令无法安装就需要直接下载对应的环境包了。
#下载地址:
https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述
在这里CRTL+F输入你需要的版本号,然后点击下载:(我在这里讲一下挑选规则)

  1. 查看自己新建的conda环境的平台是多少位的

    #输入此命令查看
    conda info
    

在这里插入图片描述
圈红的显示的是你的环境位数,也就是32还是64,如果你是64那么没问题,如果是32,你要注意,你需要重建环境,不然后面装不上torch包,原因很简单,你的平台是32位那么你的python也是32位,这样就会导致不支持安装,因为我们的环境包后缀都是64,也就是需要64位环境。

  1. 设置环境为64位方法:
conda config --env --set subdir win-64 # windows下

然后再新建个conda环境就好了。

  1. 查看环境中的python版本:
#在环境中直接输入
python

以我的为例,我是64位平台,python3.8 windows系统,那我选这个就好了:在这里插入图片描述

3.安装torch

#新环境下输入:
pip install + 刚刚下载文件的绝对路径

在这里插入图片描述
等待安装完成即可。

4.检验torch是否成功

#输入python
python 
import torch 
torch.cuda.is_available()
#如果返回true即成功

在这里插入图片描述

很多人会忽略第二步,直接搜过攻略来装torch,这样是不合适的,一定要注意第二步,避免踩坑必备!

  • 2
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值