jlzzjlzz亚洲乱熟在线播放

系統城裝機大師 - 唯一官網:www.farandoo.com!

當前位置:首頁 > 腳本中心 > htc > 詳細頁面

tensorflow:指定gpu 限制使用量百分比,設置最小使用量的實現

時間:2020-02-06來源:系統城作者:電腦系統城

在Python代碼中指定GPU


 
  1. import os
  2.  
  3. os.environ["CUDA_VISIBLE_DEVICES"] = "0"

設置定量的GPU使用量:


 
  1. config = tf.ConfigProto()
  2.  
  3. config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的顯存
  4.  
  5. session = tf.Session(config=config)

設置最小的GPU使用量:


 
  1. config = tf.ConfigProto()
  2.  
  3. config.gpu_options.allow_growth = True
  4.  
  5. session = tf.Session(config=config)

以上這篇tensorflow:指定gpu 限制使用量百分比,設置最小使用量的實現就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持我們。

分享到:

相關信息

系統教程欄目

欄目熱門教程

人氣教程排行

站長推薦

熱門系統下載