问题描述 在用别人的代码跑实验时候,需要指定GPU来跑,但是代码里面又没有给GPU编号相关参数。 解决方法 简单粗暴,直接在开头指定 CUDA_VISIBLE_DEVICES=0 python train.py 同时这也是官方建议的方法