天天看点

Ubuntu 18.04 安装 Anaconda 配置 pytorch 虚拟环境1、安装 Anaconda 2、配置pytorch虚拟环境3、测试pytorch

此为记录在服务器上配置pytorch环境步骤。

目录

1、安装 Anaconda 

2、配置pytorch虚拟环境

1、使用activate激活该环境 

2、安装pytorch

3、测试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

之后安装过程,请参见 之前安装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 remove -n pytorch  --all
           

(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的路径
           

继续阅读