Anaconda安装pytorch0.4.1和对应的torchvision

目前pytorch版本已经更新到了1.7.x了,但是还是有很多代码的环境是基于1.0以下的版本,这就给代码复现创造了难题,总所周知,配置环境是成功的一半!经过查阅资料和我的努力,终于把环境配置成功了!接下来就分享我的经验

1.先说我踩的坑!!!看了其他博客说是如下操作就行:

conda install pytorch=0.4.1 

这是在添加了镜像源的情况下(比如清华镜像源),但是这样我发现总是报错,错误显示不能找到相应的安装包,然后去了pytorch官网,发现了官网的如下代码(前提是用anaconda安装)

conda install pytorch=0.4.1 -c pytorch

看了其他解释说-c pytorch的作用是从官网地址安装,没有的话可以用镜像源安装,毕竟从官网安装的话速度很慢!
抱着试一试的态度我从官网安装,啪的一下,很快哦,找到了相应的安装包,而且速度也很快(不知道是本来就快,还是我有一个浏览器挂了梯子,还是我家网速快?)总之安装成功了!大快人心!
原因:我估摸着是安装的镜像源里不支持老版本了,还好我去官网瞄了一眼,pytorch官网地址,给了相应的版本以及安装方法。官方文档还是永远滴神!

2.安装torchvision

当然安装了pytorch,肯定还需要安装torchvision,那安装什么版本才是对应的呢?版本问题真是让人抓耳挠腮,通过查很多资料和实验很多次找到一个简单的方法:

conda install torchvision -c pytorch

anaconda会自动根据pytorch的版本帮你选择对应的torchvision,这个真是大好福利!也是,程序员为啥为难程序员呢。

3.验证安装的是否成功

进入anaconda控制台,先激活自己的虚拟环境(activate+环境名),输入 python,之后输入 import torch,如果没有报错,意味着 pytorch 已经顺利安装了,接下来输入 torch.cuda.is_available(),如果是 True,意味着可以使用 GPU训练,如果是 False,意味着只能使用CPU了。
在这里插入图片描述
温馨提示:如果是安装最新版本的话,去掉-c pytorch,毕竟用镜像源还是快一些的。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值