欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-06-20 21:52:04
如题,超级详细的靠谱的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’)查了一天无果,所以索性装...

如题,超级详细的靠谱的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)
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
创建之后: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”回车之后就是保存并退出
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
然后输入:
pip3 install torch=0.4.0 torchvision=0.4.1 -f https://download.pytorch.org/whl/cu100/torch_stable.html
这一步如果出现权限不足问题,记得在指令最后加上–user
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
安装开始,下载速度很快。
https://www.cnblogs.com/jeshy/p/11759999.html+https://blog.csdn.net/qq_18649781/article/details/89848266
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
测试
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
安装完成,测试发现报错,因为python版本不对,应该匹配,所以输入指令:conda install python=3.6
降python版本为3.6,这个过程较慢,多试几次就好了~我就试了好几次!
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
再次测试
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
成功!
*温馨提示:虽然pytorch安装好了,但是记得检查一下cuda和pytorch是不是匹配的!!!
具体方法就是,在安装的pytorch环境中输入:

python
import torch
torch.cuda.is_available()

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

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