1.no CUDA-capable device is detected
首先考虑的是cuda的驱动问题,查看gpu显示是否正常,然后更新最新的cuda驱动;
第二个考虑的是cuda设备的默认参数是否修改,平常一块显卡的设置是0,多块可能会修改此参数:
CUDA_VISIBLE_DEVICES="3" ,把它修改为0即可解决。
2.out of gpu memory
调小batch_size。
3.因为模型结构下载出现问题,或者别的预加载文件出现问题
直接换网络,或者用外网。
4.明显的存在的文件,但是导入代码出现错误,例如
from ..utils import get_upsampling_weight两点代表上上一级目录,这里识别不出来,可以直接设定根目录,然后把两个点去掉.
原文地址:https://www.cnblogs.com/ywheunji/p/10607357.html
时间: 2024-10-12 17:07:26