如何安装Tensorflow GPU版本

可以安装对应版本的cudatoolkit   cudnn

我这次需要安装tensorflow-gpu==1.15.0 经查看 对应的cuda 10

所以：

conda install cudatoolkit==10.0.130

它对应的是cudnn 7.4 但是没安装成功 

然后我直接输入 

conda install cudnn

 它根据cuda10 找到了对应的cudnn 7.6.5

安装之后

输入：

import tensorflow as tf

tf.test.is_gpu_available()

返回True 成功了。