Ubuntu使用conda新建环境(environment)并安装Tensorflow GPU版本和PyTorch

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

猜你喜欢

转载自blog.csdn.net/feifei3211/article/details/88367721