在训练yolov3的模型时,一直报错,百度了n种方法都不行
无法调用gpu而出错,但是将cuda=true改为false后就可以运行
故怀疑可能是cuda的问题,也可能是torch版本不对
我又去看了自己的环境配置,发现自己在安装pytorch的时候,后面跟了一个cudatoolkit=9.2,导致和cuda10.0不匹配,所以出问题了,无法调动gpu
重新安装后就成功了
在训练yolov3的模型时,一直报错,百度了n种方法都不行
无法调用gpu而出错,但是将cuda=true改为false后就可以运行
故怀疑可能是cuda的问题,也可能是torch版本不对
我又去看了自己的环境配置,发现自己在安装pytorch的时候,后面跟了一个cudatoolkit=9.2,导致和cuda10.0不匹配,所以出问题了,无法调动gpu
重新安装后就成功了