深度学习时GPU显存利用率高但是GPU利用率低的解决办法
代码中添加:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" //这里是自己的GPU编号，我是"0",原文是"8".
代码中添加:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" //这里是自己的GPU编号，我是"0",原文是"8".