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