keras
镜洲
小小的天有大大的梦想
展开
-
如何在基于tensorflow的深度学习框架keras中指定GPU
import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="0,1" 使用0号, 1号GPU原创 2017-05-18 10:36:57 · 4860 阅读 · 0 评论 -
如何在基于tensorflow的深度学习框架keras中指定GPU内存使用大小
set_gpu.py import os import tensorflow as tf import keras.backend.tensorflow_backend as KTF def get_session(gpu_fraction=0.3): '''Assume that you have 6GB of GPU memory and want to allocate ~2GB转载 2017-05-18 14:44:46 · 2155 阅读 · 0 评论