基于tensorflow指定GPU運行及GPU資源分配的幾種方式小結
更新時間：2020年02月03日 11:47:17 作者：青松愉快
今天小編就為大家分享一篇基于tensorflow指定GPU運行及GPU資源分配的幾種方式小結，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
1. 在終端執(zhí)行時設置使用哪些GPU（兩種方式）
(1) 如下（export 語句執(zhí)行一次就行了，以后再運行代碼不用執(zhí)行）
(2) 如下
2. 代碼中指定（兩種方式）
(1)
import os os.environ["CUDA_VISIBLE_DEVICES"] = "1"
(2)
# Creates a graph.
with tf.device('/gpu:1'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print sess.run(c)
若想使用多個GPU,如下
c = []
for d in ['/gpu:0', '/gpu:1']:
with tf.device(d):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3])
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2])
c.append(tf.matmul(a, b))
with tf.device('/cpu:0'):
sum = tf.add_n(c)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# Runs the op.
print sess.run(sum)
3.GPU資源分配
(1) 設置允許GPU增長
config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Session(config=config, ...)
(2) 設置每個GPU內存使用多少
config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.4 session = tf.Session(config=config, ...)
以上這篇基于tensorflow指定GPU運行及GPU資源分配的幾種方式小結就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Python提供了多種庫和工具,使得創(chuàng)建和定制3D曲面圖變得簡單,本文將介紹如何使用Matplotlib和mpl_toolkits.mplot3d庫繪制3D曲面圖,感興趣的可以了解下2024-04-04
這篇文章主要介紹了PyAV?基礎及使用方法分享的相關資料,需要的朋友可以參考下2023-08-08
這篇文章主要介紹了Python的joblib模型固化函數解析,joblib提供了三個與對象序列化和模型固化相關的函數hash,dump,load,joblib.hash主要是為了提供一個numpy對象的hash方法,需要的朋友可以參考下2023-08-08
這篇文章主要為大家介紹了Python全棧之協(xié)程，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-12-12
這篇文章主要介紹了keras K.function獲取某層的輸出操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
這篇文章主要介紹了Python編程判斷一個正整數是否為素數的方法,涉及Python數學運算相關操作技巧,需要的朋友可以參考下2017-04-04
jmespath是python的第三方模塊，是需要額外安裝的。它在python原有的json數據處理上做出了很大的貢獻。本文將詳細介紹如何利用jmespath實現json數據處理，需要的可以參考一下2022-03-03
循環(huán)的作用就是讓指定的代碼重復的執(zhí)行,while循環(huán)最常用的應用場景就是讓執(zhí)行的代碼按照指定的次數重復執(zhí)行,這篇文章主要給大家介紹了關于Python新手教程之while循環(huán)20例的相關資料,需要的朋友可以參考下2024-05-05
Python調用ChatGPT制作基于Tkinter的桌面時鐘
這篇文章主要為大家詳細介紹了Python如何調用ChatGPT制作基于Tkinter的桌面時鐘，文中的示例代碼講解詳細，感興趣的可以了解一下2023-03-03
大家好，本篇文章主要講的是用Python實現批量生成法務函代碼，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下2022-02-02