檢測tensorflow是否使用gpu進行計算的方式
更新時間：2020年02月03日 10:27:05 作者：匠人_C
今天小編就為大家分享一篇檢測tensorflow是否使用gpu進行計算的方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
如下所示：
import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
查看日志信息若包含gpu信息，就是使用了gpu。
其他方法：跑計算量大的代碼，通過 nvidia-smi 命令查看gpu的內存使用量。
以上這篇檢測tensorflow是否使用gpu進行計算的方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了詳解python里使用正則表達式的分組命名方式的相關資料,希望通過本文能幫助到大家，需要的朋友可以參考下2017-10-10
邊緣檢測是圖像處理和計算機視覺當中的基本問題，邊緣檢測的目的是標識數(shù)字圖像中亮度變化明顯的點。本文將通過示例和大家介紹一下邊緣檢測的原理，希望對大家有所幫助2022-12-12
這篇文章主要介紹了Python實現(xiàn)處理管道的方法,實例分析了Python實現(xiàn)管道調用子程序的技巧,需要的朋友可以參考下2015-06-06
這篇文章主要介紹了Python實現(xiàn)圖片和base64轉換詳解,Base64是一種二進制到文本的編碼方式,如果要更具體一點的話,可以認為它是一種將 byte數(shù)組編碼為字符串的方法,而且編碼出的字符串只包含ASCII基礎字符,需要的朋友可以參考下2024-01-01
這篇文章主要介紹了Python中基本的日期時間處理的學習教程,日期時間相關模塊的使用是Python入門學習中的基礎知識,需要的朋友可以參考下2015-10-10
這篇文章主要介紹了python3 實現(xiàn)自定義切片類為左閉右閉詳情，文章圍繞主題展開詳細的內容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2021-12-12
解決Pytorch 訓練與測試時爆顯存(out of memory)的問題
今天小編就為大家分享一篇解決Pytorch 訓練與測試時爆顯存(out of memory)的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
Python3.10.4激活venv環(huán)境失敗解決方法
這篇文章主要介紹了Python3.10.4激活venv環(huán)境失敗解決方法的相關資料,需要的朋友可以參考下2023-01-01
這篇文章主要介紹了python中reader的next用法,分別介紹了python3中的用法和python2中的用法，具體實例代碼大家參考下本文2018-07-07
Python實現(xiàn)批量合并多個txt文件并生成Excel文件
在數(shù)據(jù)處理中,有時會面臨合并多個文本文件的任務,本文將詳細介紹如何使用Python批量合并多個txt文件,并將其生成為一個Excel文件,需要的可以參考下2023-12-12