- Anaconda管理员运行
- 显示机器上GPU情况
nvidia-smi
- 框框处为GPU编号
- 终端执行程序时指定GPU
CUDA_VISIBLE_DEVICES=0 python xxx.py
- python代码中指定GPU
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
nvidia-smi
CUDA_VISIBLE_DEVICES=0 python xxx.py
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"