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

最新推荐文章于 2025-06-16 21:01:32 发布

原创最新推荐文章于 2025-06-16 21:01:32 发布 · 2k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#环境管理 #tensorflow #Pytorch #conda

包安装专栏收录该内容

19 篇文章

订阅专栏

本文指导如何在Ubuntu系统中使用conda创建包含TensorFlow GPU版和PyTorch的环境，并配置清华源加速包下载。介绍了conda的基本操作，如环境创建、激活、包安装及pip源设置。

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

conda相关资源
新建环境并且安装包
pip设置清华源

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.youkuaiyun.com/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.youkuaiyun.com/lambert310/article/details/52412059
其他的源地址：https://zhuanlan.zhihu.com/p/62899936