我的cuda版本是9.0,cudnn版本是7.1.2,tensorflow-gpu版本是1.9.0。
这里显示的是Tesla P4,410.66 NVIDIA driver
然后查看一下文件内容:
cat /etc/modprobe.d/blacklist.conf
systemctl set-default multi-user.target
pip install h5py==2.8.0rc1
来解决。nvidia-smi python3 import tensorflow as tf sess = tf.Session()