问题:
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版权协议,转载请附上原文出处链接和本声明。