看了好多教程都提到了使用 os.environ["CUDA_VISIBLE_DEVICES"] = "1" 或者export CUDA_VISIBLE_DEVICES=1
但是我加進(jìn)代碼里面速度并沒有提高，查了很久才找到問題所在，當(dāng)你的電腦有兩塊以上GPU時(shí)，上面這兩句代碼才起作用！

因?yàn)槲业碾娔X只有一塊GPU，所以需要將參數(shù)的“1”改成“0”才可以，否則找不到“1”號設(shè)備，它就會(huì)默認(rèn)使用CPU，速度不會(huì)提高。

如果是在終端中運(yùn)行python程序，使用命令：CUDA_VISIBLE_DEVICES=0 python filename.py即可

-------------------------------------------------補(bǔ)充：---------------------------------------------
如果有多塊卡，想要指定多個(gè)gpu訓(xùn)練，可以這樣設(shè)置：

def set_gpus(gpu_index):
if type(gpu_index) == list:
gpu_index = ','.join(str(_) for _ in gpu_index)
if type(gpu_index) ==int:
gpu_index = str(gpu_index)
os.environ["CUDA_VISIBLE_DEVICES"] = gpu_index
1
2
3
4
5
6
通過調(diào)用set_gpu函數(shù)即可實(shí)現(xiàn)。

Attention：
一定要注意的是，CUDA_VISIBLE_DEVICES，每！一！個(gè)！字！母！ 必須完全寫對，如果不小心寫成了 CUDA_VISIBLE_DIVICES 或者 CUDA_VISIABLE_DEVICES ，代碼是不會(huì)報(bào)錯(cuò)的，但是GPU調(diào)用不會(huì)成功，排查問題浪費(fèi)了我好幾個(gè)小時(shí)。。。切記！
---------------------
作者：aaon22357
來源：CSDN
原文：https://blog.csdn.net/aaon22357/article/details/82733218
版權(quán)聲明：本文為博主原創(chuàng)文章，轉(zhuǎn)載請附上博文鏈接！

總結(jié)

以上是生活随笔為你收集整理的python开启GPU加速的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。