日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

python沒有gpu,如何改用cpu跑代碼_python

作者:孩紙D ? 更新時間: 2022-07-09 編程語言

沒有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

cpu

gpu

gpu

簡單測試:gpu比cpu快5秒

原文鏈接:https://blog.csdn.net/weixin_42227243/article/details/124192982

欄目分類
最近更新