pycharm运行程序时指定GPU

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_第1张图片

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

pycharm运行程序时指定GPU_第2张图片
现在就能在指定的GPU上运行程序了

你可能感兴趣的:(随笔,pycharm,cuda)