代碼中添加:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" //這裏是自己的GPU編號，我是"0",原文是"8".
深度學習時GPU顯存利用率高但是GPU利用率低的解決辦法
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
代碼中添加:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" //這裏是自己的GPU編號，我是"0",原文是"8".