Ubuntu使用conda新建環境(environment)並安裝Tensorflow GPU版本和PyTorch

conda相關資源

conda詳細文檔：https://docs.conda.io/projects/conda/en/4.6.1/user-guide/getting-started.html
沒有安裝conda命令可以從 https://docs.anaconda.com/anaconda/install/ 下載並且安裝。
conda可以用於創建Python環境，安裝包，各個環境的Pyhton包相互獨立。使用是激活安裝的環境即可。不激活則使用默認的環境。
下面介紹的是簡單創建一個環境並安裝包，詳細的使用說明請看conda文檔。

新建環境並且安裝包

清華源的添加見：https://blog.csdn.net/feifei3211/article/details/80361227#commentBox
安裝中假設已經安裝好了cuda環境和conda。

# 加上源用於快速獲取相關的包
# 清華源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
# 騰訊源
#conda config --add channels https://mirrors.cloud.tencent.com/anaconda/pkgs/free/
#conda config --add channels https://mirrors.cloud.tencent.com/anaconda/pkgs/main/
#conda config --add channels https://mirrors.cloud.tencent.com/anaconda/cloud/conda-forge/
#conda config --add channels https://mirrors.cloud.tencent.com/anaconda/cloud/pytorch/
#conda config --add channels https://mirrors.cloud.tencent.com/anaconda/pkgs/pro/

# 創建一個環境，相關的包存在自己目錄下（以後激活環境的時候只要指定這個目錄即可）
# --prefix 指定目錄的位置(自己指定即可), python=3.6默認的python版本
# 使用-n（--name）指定的話可能會出現權限問題，因爲這默認安裝在conda的環境目錄下面。
conda create --prefix ./py3test python=3.6

# 激活環境，之後安裝的包都會在這個環境中。
# 恢復默認的環境：source deactivate
source activate ./py3test

# 安裝tensorflow-gpu和pytorch，確定安裝即可。（只用包名tensorflow不帶GPU支持）
# 安裝想要的包用conda install package_name，
conda install tensorflow-gpu pytorch

# 測試安裝的包的版本號
python -c 'import tensorflow as tf; print(tf.__version__); import torch; print(torch.__version__)'

pip設置清華源

有時conda可能沒有相關的包，這是用pip安裝可能速度較慢，建議添加pip的清華源。只要激活了相關的環境，用pip和conda安裝python包都會安裝到激活到環境中。沒激活安裝到默認到python環境中。添加方式如下

mkdir -p ~/.pip
vim ~/.pip/pip.conf
寫入如下
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple' >> 

參考:
具體添加方法：https://blog.csdn.net/lambert310/article/details/52412059
其他的源地址：https://zhuanlan.zhihu.com/p/62899936