(转载)ubuntu 18.04 安装pytorch

第一步:安装pip3

sudo apt-get install python3-pip

顺便安装numpy

pip3 install numpy

然后会提示你更新pip的版本,于是我就选择更新了

pip3 install --upgrade pip

第二步:配置pip使用的镜像源


附:

国内镜像源列表

豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/


参照经验贴,采用ailiyun镜像

运行命令:

cd ~/.pip

如果不存在该文件夹则:

mkdir ~/.pip
cd ~/.pip
touch pip.conf
gedit pip.conf

此时会跳出pip.conf窗口,在pip.conf中添加以下内容:

[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com

点击右下角的保存即可
第三步:安装pytorch

运行命令:

pip install torch torchvision

下载速度还挺快的,但是download完成之后提示安装指令的权限不够(忘记截图了),提示如下:

Could not install packages due to an EnvironmentError: Permission denied: ‘/usr/local/lib/python3.5/dist-packages/pylab.py’ Consider using the --user option or check the permissions.

加上–user真的就成功了,运行命令:

pip install torch torchvision --user

结果如图:

第四步:测试pytorch是否安装成功

在终端输入

python3(注意是输入python3,如果只输入python,会默认是python2,当然,之后你也可以自己更改python的默认版本)

import torch

import torchvision

结果如图:

显示如图应该就算安装成功了。

希望对你有帮助。

————————————————
版权声明:本文为CSDN博主「番茄很甜」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_42165939/article/details/86181585