- 官方这样说道
- 如果是tensorflow、cntk作为后端,那么会自动检测
- theano可以手动设置
但是我这的速度,让人怀疑并没有运行在gpu上(一张图,运行时间近10秒)
-
尝试运行时指定
CUDA_VISIBLE_DEVICES=0 python demo.py
运行时间无变化 -
显式设置backend
import keras.backend.tensorflow_backend as KTFimport keras.backend.tensorflow_backend as KTF
运行时间无变化 -
tensorflow日志级别控制
impot os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
"""
TF_CPP_MIN_LOG_L