在项目的Run/debug configuration进行如下设置:
Environment:

CUDA_VISIBLE_DEVICES=4,5
#这是设置4,5号显卡可见。
最后修改:2023 年 05 月 15 日
如果觉得我的文章对你有用,请随意赞赏