Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤

2022-08-10,,,,

如题,超级详细的靠谱的Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0教程!!!
因为最近在训练的网络RCAN是pytorch0.4.0版本的代码,在pytorch1.x版本运行总是出错,(比如AttributeError: ‘_MSDataLoaderIter’ object has no attribute '_put_indices’或者ImportError: cannot import name ‘_DataLoaderIter’)查了一天无果,所以索性装了一个pytorch0.4.0,下面分享一下我在Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0的历程:
首先,新建一个想要配置的环境的名称:conda crate -n (name)

创建之后:source activate
进入环境,开始配置环境:
先:cd ~/.pip
如果没有该文件夹,就创建一个:mkdir ~/.pip
再次进入:cd ~/.pip
touch pip.conf
vim pip.conf #进入文件并开始编辑文件
输入”i”开始编辑:
[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com

复制进去之后按”Esc”->”:wq”->”Enter”回车之后就是保存并退出

然后输入:
pip3 install torch=0.4.0 torchvision=0.4.1 -f https://download.pytorch.org/whl/cu100/torch_stable.html
这一步如果出现权限不足问题,记得在指令最后加上–user

安装开始,下载速度很快。
https://www.cnblogs.com/jeshy/p/11759999.html+https://blog.csdn.net/qq_18649781/article/details/89848266

测试

安装完成,测试发现报错,因为python版本不对,应该匹配,所以输入指令:conda install python=3.6
降python版本为3.6,这个过程较慢,多试几次就好了~我就试了好几次!

再次测试

成功!
*温馨提示:虽然pytorch安装好了,但是记得检查一下cuda和pytorch是不是匹配的!!!
具体方法就是,在安装的pytorch环境中输入:

python
import torch
torch.cuda.is_available()

看输出是否是True!
(本人也是新人,希望大家多多批评指正)

本文地址:https://blog.csdn.net/weixin_44934094/article/details/107127539

《Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤.doc》

下载本文的Word格式文档,以方便收藏与打印。