Environment Variable CUDA_VISIBLE_DEVICES is not set correctly. If you wanna use gpu, please set CUDA_VISIBLE_DEVICES as cuda_device_id.
增加设置
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
版权声明：本文为qq_15821487原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。