pycharm运行程序时指定GPU

welcome to my blog

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

1. 在源代码中添加

import os
# 指定使用0,1,2三块卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2" 

2. 使用命令行启动程序时加上CUDA_VISIBLE_DEVICES=0,1,2
比如, CUDA_VISIBLE_DEVICES=0,1,2 python FasterRCNN.py

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

第一步, 打开"Edit Configurations"

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

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