python沒有gpu,如何改用cpu跑代碼
更新時間:2022年05月16日 14:38:40 作者:孩紙D
這篇文章主要介紹了python沒有gpu,如何改用cpu跑代碼?今天小編就為大家分享一下解決方案。如有錯誤或未考慮完全的地方,望不吝賜教
沒有gpu,改用cpu跑代碼
以為很麻煩的修改,很大動干戈的改動,以及無止盡的報錯,結果,就改動代碼中的device就OK了,驚不驚喜,意不意外???
也就是說,
只需要將device設置為以下即可
device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu"))
另外也不需要全如此設置,只需查看報錯處調用device的來源,在修改torch.device()括號里面的內容就可。
Tensorflow-gpu禁用gpu設置(cpu與gpu速度對比)
禁用gpu設置
# 在import tensorflow之前 import os os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
cpu與gpu對比
顯卡:GTX 1066
cpu
gpu
簡單測試:gpu比cpu快5秒
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Python實現動態(tài)給類和對象添加屬性和方法操作示例
這篇文章主要介紹了Python實現動態(tài)給類和對象添加屬性和方法操作,涉及Python面向對象程序設計中類與對象屬性、方法的動態(tài)操作相關實現技巧,需要的朋友可以參考下2020-02-02在Windows服務器下用Apache和mod_wsgi配置Python應用的教程
這篇文章主要介紹了在Windows服務器下用Apache和mod_wsgi配置Python應用的教程,本文選擇以flask框架作為示例,需要的朋友可以參考下2015-05-05