可以用如下代码检查tensorflow版本是否可用GPU
import tensorflow as tf
tf.test.is_gpu_available()
tf.test.is_built_with_cuda()
由于没有安装tensorflow-gpu,使用keras无法利用GPU训练。需要安装tensorflow-gpu和keras
可以用如下代码检查tensorflow版本是否可用GPU
import tensorflow as tf
tf.test.is_gpu_available()
tf.test.is_built_with_cuda()
由于没有安装tensorflow-gpu,使用keras无法利用GPU训练。需要安装tensorflow-gpu和keras