这是GPU内存耗尽的结果;
解决方法:不使用GPU,直接使用CPU;
开始尝试:CUDA_VISIBLE_DEVICES=0 python filename.py
效果依旧;
然后尝试:os.environ["CUDA_VISIBLE_DEVICES"]="-1";就OK了
这是GPU内存耗尽的结果;
解决方法:不使用GPU,直接使用CPU;
开始尝试:CUDA_VISIBLE_DEVICES=0 python filename.py
效果依旧;
然后尝试:os.environ["CUDA_VISIBLE_DEVICES"]="-1";就OK了