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了。
Anaconda安装pytorch0.4.1和对应的torchvision_第1张图片
温馨提示:如果是安装最新版本的话,去掉-c pytorch,毕竟用镜像源还是快一些的。

你可能感兴趣的:(软件及环境安装,anaconda,python,pytorch,深度学习)