- 博客(2)
- 收藏
- 关注
转载 如何在基于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
原创 如何在基于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
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人