可以用如下代码检查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
4657
1万+

被折叠的 条评论
为什么被折叠?