指定python运行使用GPU

  • Anaconda管理员运行
  • 显示机器上GPU情况
nvidia-smi
  • 框框处为GPU编号
    指定python运行使用GPU_第1张图片
  • 终端执行程序时指定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"

你可能感兴趣的:(为吃饭学着拿筷子,python,tensorflow_gpu)