目前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
,毕竟用镜像源还是快一些的。