Ubuntu(GPU)和MacOS(CPU) 在Conda沙盒环境下安装PyTorch 1.0.0

沙盒环境的好处在于它是独立的，不会影响其他的环境。
PyTorch 1.0.0是Facebook发布的框架，通过它可以很容易地迁移Mask R-CNN以及其它的开源项目。
安装GPU的话，你首先要保证机器的GPU已经配置完成，教程链接。
注意：我使用的是Ubuntu 16.04 + CUDA 9.0 + CuDNN 7.1.4，CPU基于MacOS 10.14.1

1 创建Pytorch 1.0.0环境

注意：这里我安装的是python 3.6.5的环境，可以根据自己需求进行修改

conda create --name pytorch-1.0.0 python=3.6.5

2 管理环境的基础命令

2.1 激活环境

source activate pytorch-1.0.0

2.2 关闭当前激活的环境

source deactivate

2.3 移除环境

conda remove --name pytorch-1.0.0 --all

2.4 查看当前的环境

conda info --envs

2.5 重命名

clone需要重命名的环境，然后删除旧环境

conda create --name new_name --clone old_name
conda remove --name old_name --all

3 安装PyTorch 1.0.0

进入PyTorch，点击Preview 并选择 OS、 Package、Language、CUDA，复制页面上给出的命令。激活当前环境再进行安装

3.1 Ubuntu 16.04

source activate pytorch-1.0.0
conda install pytorch torchvision -c pytorch

3.2 MacOS 10.14.1

source activate pytorch-1.0.0
conda install pytorch torchvision -c pytorch

简单验证一下

python -c "import caffe2; import torch; import torchvision"

4 安装其它的库

以下两种方法conda和pip均适用，根据自己喜好选择就好。这里以安装opencv、jupyter和tqdm为例。推荐使用先激活再安装的方式

4.1 base环境下直接制定环境进行安装

conda install --name pytorch-1.0.0 opencv jupyter
pip install -t /home/czy/miniconda3/envs/pytorch-1.0.0/lib/python3.6/site-packages/ tqdm

4.2 先激活环境再安装

source activate pytorch-1.0.0
conda install opencv jupyter
pip install tqdm

5 Notebook和PyCharm使用沙盒环境

5.1 Jupyter Notebook

先激活当前的环境，再打开Notebook

source activate pytorch-1.0.0
jupyter notebook

5.2 PyCharm

载入项目时，选择该项目的编译环境为当前沙盒环境即可。

Reference

Conda 官网
PyTorch 官网