江南烟雨尘

摘要： 首先指定GPU： 两种方法：1.在命令行指定：CUDA_VISIBLE_DEVICES=3,0,1,2 python main.py 也就是在普通的python main.py 命令前加 CUDA_VISIBLE_DEVICES=gpu编号 注意：等号两边不能有多余的空格 2.在main.py最前面 阅读全文