1.首先需要確認是否成功安裝cuda,代碼見圖一;打印結(jié)果如圖二所示。
?圖一
?文章來源地址http://www.zghlxwxcb.cn/news/detail-572790.html
圖二?
2.如果未安裝成功可以自行搜索,不麻煩;安裝成功后需要分三步設(shè)置使用GPU,以簡單的softmax分類器為例:
a.導(dǎo)入os模塊
import os #指定參與運算的顯卡為GPU1,這個需要自己進系統(tǒng)管理器查看自己的顯卡是GPU序號 os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
b.將模型放進GPU中運算。
?c.更改訓(xùn)練、測試兩個步驟,使用GPU運算。
?文章來源:http://www.zghlxwxcb.cn/news/detail-572790.html
?
到了這里,關(guān)于python運行使用gpu運算【python基礎(chǔ)】python開啟GPU加速的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!