Tensorflow 2.0.0-alpha 安装 Linux系统

时间:2021-11-13 07:22:07

1、TensorFlow2.0的安装测试 Linux

Tensorflow Dev Summit 正式宣布 Tensorflow 2.0 进入 Alpha 阶段。

基于 Anaconda 创建环境一个尝鲜环境:

conda create -n tf2

然后:

pip install tensorflow-gpu=2.0.0-alpha

由于 2.0 依赖的是 CUDA 10 所以系统内多半没有,直接用 conda 在虚拟环境装一个:

conda install cudatoolkit cudnn

这个安装后并不在 Tensorflow 预期的位置上,所以正常需要处理一下:

export LD_LIBRARY_PATH=~/anaconda3/envs/tf2/lib

然后就可以运行 python 检验一下:

Tensorflow 2.0.0-alpha 安装 Linux系统