在代码的开头,所有 import 之前,加入下面两行代码:import osos.environ["CUDA_VISIBLE_DEVICES"] = "-1" # 这一行注释掉就是使用cpu,不注释就是使用gpu。