tensorflow設定gpu及gpu視訊記憶體使用

2021-08-19 19:23:25 字數 596 閱讀 9858

os.environ[

'cuda_visible_devices'

] = 

'0'#使用 gpu 0

os.environ['cuda_visible_devices'

] = 

'0,1'

# 使用 gpu 0,1

一是讓tensorflow在執行過程中動態申請視訊記憶體,需要多少就申請多少;第二種方式就是限制gpu的使用率。

config = tf.configproto()  

config.gpu_options.allow_growth = true

session = tf.session(config=config) 

(2) 在tensorflow中定義session時作如下設定,該設定會強制程式只占用指定比例的gpu視訊記憶體。 

gpu_options=tf.gpuoptions(per_process_gpu_memory_fraction=

0.4)  

config=tf.configproto(gpu_options=gpu_options)  

session = tf.session(config=config)  

tensorflow設定GPU佔用量

參考 tensorflow設定gpu佔用量 tensorflow在訓練時預設占用所有gpu的視訊記憶體。可以通過以下方式解決該問題 1 在構造tf.session 時候通過傳遞tf.gpuoptions作為可選配置引數的一部分來顯式地指定需要分配的視訊記憶體比例,如下所示 假如有12gb的視訊記憶體...

TensorFlow設定GPU佔用量

tf.configproto一般用在建立session的時候。用來對session進行引數配置 with tf.session config tf.configproto tf.configproto 的引數 log device placement true 是否列印裝置分配日誌 allow so...

Tensorflow全域性設定可見GPU編號

筆者需要tensorflow僅執行在乙個gpu上 機器本身有多gpu 而且需要依據系統引數動態調節,故無法簡單使用cuda visible devices。一種方式是全域性使用tf.device函式生成的域,但裝置號需要在繪製graph前指定,仍然不夠靈活。查閱文件發現config的gpuoptio...