TensorFlow1.2~2.1各GPU版本与CUDA对应版本|简记


参考链接一
参考链接二


搭建TensorFlow的GPU版本,必备条件是一块能够支持CUDA的NVIDIA显卡,首先需要安装其基础支持平台CUDA和其机器学习库cuDNN,然后在此基础上搭建对应TensorFlow GPU版本

TensorFlow1.2~2.1各GPU版本CUDA和cuDNN对应版本如下:
TensorFlow1.2~2.1各GPU版本与CUDA对应版本|简记_第1张图片


测试:
Cuda 10.0, V10.0.130
首先 安装了:

conda create -n tf21 python=3.7.2

pip install tensorflow-gpu==2.1.0

测试输出如下:
TensorFlow1.2~2.1各GPU版本与CUDA对应版本|简记_第2张图片
import 时 说 不能加载 cuda下的一个.so 文件,但是 print(tf.__version__) 可以顺利输出。

因为 cuda版本为 10.0, 担心 代码运行出问题,所以还是 进行卸载,重新安装了 2.0.0
命令如下:
pip源设置

pip uninstall tensorflow-gpu==2.1.0

pip install tensorflow-gpu==2.0.0

pip 设置使用豆瓣 或者 阿里云的 源,下载速度还是可以的!
安装后再次进行测试,就是完全正确的了,没有刚刚多余的输出!

Python 3.7.2 (default, Dec 29 2018, 06:19:36) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> print(tf.__version__)
2.0.0



你可能感兴趣的:(TensorFlow)