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

标签：默认   设备   pytorch   导入   dev   cte   识别   ble   模型   

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

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

标签：默认   设备   pytorch   导入   dev   cte   识别   ble   模型   

原文地址：https://www.cnblogs.com/ywheunji/p/10607357.html