Vscode 配置debug模式下 指定gpu

问题:

Vscode debug模式下的默认gpu是0,即使代码里面配置了,debug起来还是会使用默认的0号gpu,现修改配置如下,实现任意指定gpu。

方法一:

.py中指定gpu

 import os
 os.environ['CUDA_VISIBLE_DEVICES'] = '1'

方法二:

第一步:
运行 -> 添加配置->Python
在这里插入图片描述
在这里插入图片描述

第二步:
添加配置

 "env": {
     "CUDA_VISIBLE_DEVICES": "1"
 }

在这里插入图片描述
第三步:
测试通过~
在这里插入图片描述


版权声明:本文为qq_40662074原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。