keras gpu的问题

官方这样说道 如果是tensorflow、cntk作为后端,那么会自动检测 theano可以手动设置 但是我这的速度,让人怀疑并没有运行在gpu上(一张图,运行时间近10秒) 尝试运行时指定 CUDA_VISIBLE_DEVICES=0 python demo.py 运行时间无变化 显式设置backend import keras.backend.tensorflow_backend as KTF
相关文章
相关标签/搜索