可以安装对应版本的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 成功了。