原文:
Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
tensorflow各个版本的CUDA以及Cudnn版本对应关系
Ubuntu16.04 + CUDA9.0 + cuDNN7.3 + Tensorflow-gpu-1.12 + Jupyter Notebook 深度学习环境配置
Linux端:https://tensorflow.google.cn/install/source
图片.png 图片.png
图片.png
图片.png
图片.png
图片.png 图片.png
安装tensorflow-gpu版本
直接使用sudo pip install tensorflow-gpu 命令安装(前提已经安装pip 或pip3),这时会发现安装的是最新版本,有可能与cuda版本不匹配。import tensorflow也会报错。可以下载一个gpu版本的tensorflow,离线在本地使用pip安装。(下载地址:http://pypi.org/project/tensorflow-gpu/1.12.0/#files)
sudo pip install ******(下载的文件)
安装tensorflow-gpu版本(如果能上外网)
pip install tensorflow-gpu==1.10
网友评论