linux安装TensorFlow-GPU版本

一、安装CUDA和cuDNN

1.CUDA和cuDNN的版本对应关系

去官网查看：https://developer.nvidia.com/rdp/cudnn-archive

2.TensorFlow与CUDA、cuDNN的版本对应关系

https://tensorflow.google.cn/install/source#linux

 

 

 

 

二、根据对应的版本安装TensorFlow-gpu版本

# 创建虚拟环境：tmp_python
conda create -n tmp_python python=2.7

# 显示自己创建的虚拟环境
conda env list
# 进入创建虚拟环境
source activate tmp_python

#安装指定 版本的TensorFlow
pip install tensorflow-gpu==1.9.0

# 安装的包
conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.1.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
 
注意：
对应版本的cudatoolkit和cudnn安装不好时，可以在安装tensorflow-gpu之后直接输入一下命令
conda install cudnn　　　　#不加版本号

 

参考文献：

①https://www.jianshu.com/p/3d8a444ff440

②https://blog.csdn.net/qq_35614920/article/details/79116103?spm=1001.2101.3001.4242