这是我参考的原文,只是个别地方添加或修改。
https://blog.csdn.net/tuyim7124/article/details/80723997
一、目标环境
ubuntu16.04+python3.6+pytorch(CPU版)
二、安装步骤
(一)安装anaconda
1.在 anaconda官网 选择 Python 3.6 version 下载,考虑到服务器在国外,下载速度会非常慢,故可选择 清华大学开源软件镜像站 下载,选择Anaconda3-5.2.0-Linux-x86_64.sh 这一版本。
2.ubuntu默认下载地址为 /home/name/下载(注:name为本机用户名),进入终端。或者在下载页面下鼠标右键,选择在终端打开.
cd /home/name/下载
bash Anaconda3-5.2.0-Linux-x86_64.sh
3.在安装过程中,有一步选择是否将环境变量添加到.bashrc中,此处选择 yes。之后一步选择是否安装VSCode,根据个人需要,如仅配置pytorch环境,则选择no即可。
4.重新进入终端,输入
python
若出现
Python 3.6.5 |Anaconda, Inc.| (default, Apr 29 2018, 16:14:56)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
则表明python3.6安装成功。
退回终端,输入
conda info
此时可能会提示 conda未找到命令,原因是虽然我们在anaconda安装过程已将其加入到系统环境变量中,但需要重启环境变量才能生效,在终端,输入
source .bashrc
之后再重试
conda info
或者打开一个新的终端,直接输入
conda info
即可查看anaconda的安装信息。
(二)安装pytorch
1.进入pytorch官网,根据系统配置选择合适的版本
由于是CPU版本,所以在CUDA那一项选择None,在终端输入
conda install pytorch-cpu torchvision-cpu -c pytorch
需要确定的地方选择yes即可,等待安装结束。
三、检测
安装结束后,重新进入终端,输入
name:~$ python
Python 3.6.5 |Anaconda, Inc.| (default, Apr 29 2018, 16:14:56)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch as t
>>> t.Tensor(3,4)
tensor([[2.2777e-26, 3.0639e-41, 2.8026e-44, 0.0000e+00],
[ nan, 3.0639e-41, 1.3733e-14, 4.7429e+30],
[1.9431e-19, 4.7429e+30, 5.0938e-14, 0.0000e+00]])
若无报错信息,就代表安装成功了,快用pytorch搞事情吧!