欢迎关注博主的公众号:happyGirl的异想世界。有更多干货还有技术讨论群哦~ 指定运行代码的GPU卡号 import os os.environ['CUDA_VISIBLE_DEVICES'] = '2' 多卡需要自己再写一下函数