win10+python3.8+pytorch安装(避坑)

win10+python3.8+pytorch安装(避坑)

安装gpu版本的pytorch

1.首先查看自己电脑是否支持

桌面右键–NVIDIA控制面板–帮助–系统信息–组件
在这里插入图片描述
这里我电脑最高支持的cuda版本是11.0.2

2、安装cuda&cudnn(也有大佬说不用安装,在安装pytorch时会自动安装)

直接下载安装就行
安装对应的cudnn
注意配置环境
这里推荐https://www.cnblogs.com/guan-zl/p/12986253.html的安装教程(pytorch安装没有采用该教程)

3、直接去pytorch官网下载相应版本

https://pytorch.org/get-started/previous-versions/
在这里插入图片描述

下载安装时,最好以管理员身份运行cmd,参考https://blog.csdn.net/iiiliuyang/article/details/123106024
**在官网提供的安装命令pip 和install之间加上–no-cache-dir install ,在命令结尾加上–user,**下载过程有点慢。
(这一步主要是为了防止在下载安装pytorch中电脑出现死机的状况)
在这里插入图片描述

Tip:

网上很多教程是用清华源镜像安装的,但是我在安装过程中发现torch.cuda.is_available()命令结果始终为FALSE,查看安装的包发现,清华源镜像给我安装的是cpu0版本的pytorch。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值