​​welcome to my blog​​

指定GPU的方式大概有两种,

  1. 在源代码中添加
import os
# 指定使用0,1,2三块卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2"
  1. 使用命令行启动程序时加上​​CUDA_VISIBLE_DEVICES=0,1,2​​​​比如, CUDA_VISIBLE_DEVICES=0,1,2 python FasterRCNN.py​​

使用pycharm调试程序时不想改动源代码, 可以这么做, 从而指定运行时使用的GPU, 只需两步

第一步, 打开"Edit Configurations"

pycharm运行程序时指定GPU_cuda

第二步, 设置环境变量, 按下图序号操作即可

pycharm运行程序时指定GPU_pycharm_02


现在就能在指定的GPU上运行程序了