快捷導(dǎo)航

使用Tensorflow-GPU禁用GPU設(shè)置(CPU與GPU速度對比)

更新時間：2020年06月30日 09:08:48 作者：ArchonGum

這篇文章主要介紹了使用Tensorflow-GPU禁用GPU設(shè)置(CPU與GPU速度對比)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

禁用GPU設(shè)置

# 在import tensorflow之前
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

CPU與GPU對比

顯卡：GTX 1066

CPU

GPU

簡單測試：GPU比CPU快5秒

補充知識：tensorflow使用CPU可以跑（運行），但是使用GPU卻不能用的情況

在跑的時候可以讓加些選項：

with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))

其中allow_soft_placement能讓tensorflow遇到無法用GPU跑的數(shù)據(jù)時，自動切換成CPU進行。

log_device_placement則記錄一些日志。

以上這篇使用Tensorflow-GPU禁用GPU設(shè)置(CPU與GPU速度對比)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python中pass的作用與使用教程
這篇文章主要給大家介紹了關(guān)于Python中pass的作用與使用教程，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-11-11
Python Tkinter模塊 GUI 可視化實例
今天小編就為大家分享一篇Python Tkinter模塊 GUI 可視化實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-11-11
Python光學仿真數(shù)值分析求解波動方程繪制波包變化圖
這篇文章主要為大家介紹了Python光學仿真通過數(shù)值分析求解波動方程并繪制波包變化圖的示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助
2021-10-10
基于Linux系統(tǒng)中python matplotlib畫圖的中文顯示問題的解決方法
下面小編就為大家?guī)硪黄贚inux系統(tǒng)中python matplotlib畫圖的中文顯示問題的解決方法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2017-06-06
python實現(xiàn)的MySQL增刪改查操作實例小結(jié)
這篇文章主要介紹了python實現(xiàn)的MySQL增刪改查操作,結(jié)合實例形式總結(jié)分析了Python基本的mysql增刪改查及銀行賬號查詢等相關(guān)操作實現(xiàn)技巧,需要的朋友可以參考下
2018-12-12
python異常處理并調(diào)試
這篇文章主要介紹了python異常處理并調(diào)試，異常是錯誤出現(xiàn)時，可以在正常的控制流程之外采取的行為下面我們就來看看python的那些異常，需要的小伙伴可以參考一下
2022-02-02
Python合并兩個字典的常用方法與效率比較
這篇文章主要介紹了Python合并兩個字典的常用方法與效率比較,實例分析并對比了Python合并字典的常用方法,需要的朋友可以參考下
2015-06-06
在django中使用自定義標簽實現(xiàn)分頁功能
這篇文章主要介紹了在django中使用自定義標簽實現(xiàn)分頁功能,具有一定的參考價值，感興趣的小伙伴們可以參考一下
2017-07-07
更換Django默認的模板引擎為jinja2的實現(xiàn)方法
今天小編就為大家分享一篇更換Django默認的模板引擎為jinja2的實現(xiàn)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-05-05
python3 小數(shù)位的四舍五入(用兩種方法解決round 遇5不進)
這篇文章主要介紹了python3 小數(shù)位的四舍五入(用兩種方法解決round 遇5不進)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-04-04