Linux安装Anaconda3并创建虚拟环境来安装tensorflow-gpu和tensorflow-federated

最新推荐文章于 2023-09-27 10:02:45 发布

欧恩方

最新推荐文章于 2023-09-27 10:02:45 发布

阅读量684

点赞数 1

分类专栏：大数据联邦学习文章标签： tensorflow 机器学习联邦学习分布式计算 anaconda

本文链接：https://blog.csdn.net/sinat_33856143/article/details/105679299

版权

大数据同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

联邦学习

4 篇文章 2 订阅

订阅专栏

这篇博客介绍了如何在Linux系统中通过Anaconda3安装TensorFlow GPU版，并创建名为'tff'的虚拟环境。接着，文章详细说明了如何添加清华源，然后使用pip在环境中安装TensorFlow Federated。最后，作者进行了测试以确保安装成功。

摘要由CSDN通过智能技术生成

安装Anaconda

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.02-Linux-x86_64.sh

bash Anaconda3-2020.02-Linux-x86_64.sh

创建并激活虚拟环境(name:tff)

conda create -n tff
conda activate tff

安装tensorflow-gpu

# conda activate tff
conda install tensorflow-gpu==2.1.0

添加清华源

mkdir ~/.pip
touch ~/.pip/pip.conf

在.pip/pip.conf中添加清华源

[global]
timeout=6000
index-url=https://pypi.tuna.tsinghua.edu.cn/simple

我们也给conda添加清华源吧

# conda activate tff
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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/msys2/

安装tensorflow-federated

由于conda里面没有tensorflow-federated，所以只能pip安装

# conda activate tff
pip install tensorflow-federated

测试

进入python shell

# conda activate tff
# python
import tensorflow as tf
import tensorflow_federated as tff
print(tf.version.VERSION) # '2.1.0'
print(tff.version.__version__) # '0.13.1'