# 使用0, 2, 3三塊GPU
os.environ['CUDA_VISIBLE_DEVICES'] = '0, 2, 3'
#設置每個GPU應該拿出多少容量給進程使用,0.6表示60%
gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.6)
config=tf.ConfigProto(
gpu_options=gpu_options,
log_device_placement=True, # log_device_placement是否打印設備分配日誌
allow_soft_placement=True) # allow_soft_placement指定設備不存在的話,是否允許TF自動分配設備
# 使用allow_growth option,開始分配少量顯存,然後按需增加, 一般和tf.GPUOptions選擇一個使用
# config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
tf.ConfigProto() 和tensorflow的GPU配置
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
ImportError: libcudnn.so.7: cannot open shared object file錯誤以及非root用戶安裝cudnn
MirrorN
2020-07-07 23:30:35
[深度之眼]LeNet/AlexNet/VGGNet/InceptionNet/ResNet實現fashion_mnist分類
TF_Keras
2020-07-08 11:53:37
Tensorflow常用函數(二)
wf592523813
2020-07-08 08:32:10
利用TensorBoad可視化網絡
小敏偏头痛
2020-07-08 06:40:26
GPU導入模型非常緩慢的解決辦法
还是少年呀
2020-07-08 05:59:41
windows 下 keras.utils.plot_model報錯問題
DDDIIILLL
2020-07-08 05:35:28
subprocess.py報錯:FileNotError: [Errno 2] No such file or directory: java: java
清晨的光明
2020-07-08 02:37:26
[tensorflow2]五分鐘帶你入門tensorboard&&錯誤解決方案
still_learning
2020-07-08 02:30:48
Mac安裝Tensorflow,運行項目報錯: module compiled against API version 0xa but this version of numpy is 0x9
铁真木
2020-07-08 01:58:44
損失函數:categorical_crossentropy
Stealers
2020-07-08 01:15:40
微軟三維人臉重建論文前期知識鋪墊——《Accurate 3D Face Reconstruction with Weakly-Supervised Learning》
wxshan3
2020-07-08 01:13:07
《Tensorflow中文社區教程》筆記
ME_Seraph
2020-07-08 00:56:51
[blog9]ubuntu18.04配置pytorch和tensorflow jupyter ipython
Drinks_
2020-07-07 23:07:48
TFRecord 的創建與讀取
苍桧丹枫
2020-07-07 21:59:47