Ubuntu16.04安装TensorFlow

时间:2023-03-09 07:06:27
Ubuntu16.04安装TensorFlow

1、查看tensoflow与CUDA对应版本:

windows端:https://tensorflow.google.cn/install/source_windows

Linux端:https://tensorflow.google.cn/install/source,对应如下图

Ubuntu16.04安装TensorFlow

2、使用清华大学开源镜像进行安装:

CPU 版本:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow == 指定版本(如1.10)

GPU 版本: 

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu == 指定版本(如1.10)