此为记录在服务器上配置pytorch环境步骤。
目录
1、安装 Anaconda
此次任务需要用 python 2,所以安装 Anaconda 的 Python 2.7 版本。具体安装过程与 之前安装Anaconda 3 无异。
不过还是推荐在官网上 https://www.anaconda.com/download/#linux 右击 Python 2.7 version 下的 Download (Linux 64位版本), 选择复制链接地址,获取下载链接后,在线安装,在终端输入:
wget https://repo.anaconda.com/archive/Anaconda2-2019.07-Linux-x86_64.sh
( ubuntu:
wget https://repo.anaconda.com/archive/Anaconda3-2020.02-Linux-x86_64.sh
)
之后安装过程,请参见 之前安装Anaconda 3 。
2、配置pytorch虚拟环境
conda create -n pytorch python=2.7
1、使用activate激活该环境
source activate pytorch
或
conda activate pytorch
激活后,会看到前面会显示环境名字比如(pytorch)。可确认python和pip的路径是否在envs下面:
which python
which pip
[Update(2020-12-10):
(1) 删除该环境
conda env remove -n pytorch
(2) 查看环境信息, 带星号*的是当前的环境
conda info --env
(Windows里面可以:conda-env list )
(3) Windows 中激活或退出环境
激活:activate pytorch
退出:deactivate
(比Ubuntu简略一点)
Done.]
2、安装pytorch
这是最关键的一步,有人推荐在这里指定固定的版本,如cuda8.0、python3.6对应版本:conda install pytorch torchvision cuda80 -c pytorch 。我依稀记得服务器上cuda10.0,但又不想费时间去确定,随将这个问题扔给了conda,让它自己选择。
conda install pytorch torchvision
若报错或失败,则可
conda install pytorch torchvision -c pytorch
最后的-c pytorch参数的意思是强制在pytorch官网下载。有时候因为墙的缘故,可能会因为这一参数导致下载失败。
(torchvision 很有用,所以,将 torchvision 和 pytorch 一同安装。)
也有用pip或源码安装的,但那样更麻烦,随弃。
3、测试pytorch
在终端打开 python 后:
import torch
import torchvision
print(torch.__version__) #输出PyTorch版本
print(torch.__path__) #输出PyTorch的路径