运行pytorch代码遇到的error解决办法

Posted 2021-11-22 ywheunji

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
两点代表上上一级目录,这里识别不出来,可以直接设定根目录,然后把两个点去掉.