Open philipclaesson opened 5 years ago
I had same issue. Downgrading torch to 0.3.1 works for me.
I had same issue. Downgrading torch to 0.3.1 works for me.
After Downgrading torch to 0.3.1, and run the 'main.py' .
RuntimeError: CUDNN_STATUS_EXECUTION_FAILED
How to solve?
I had same issue. Downgrading torch to 0.3.1 works for me.
After Downgrading torch to 0.3.1, and run the 'main.py' .
RuntimeError: CUDNN_STATUS_EXECUTION_FAILED
How to solve?
This is mostly like an issue of your CUDA. For example, if you are using 2080Ti and CUDA10, pytorch 0.3 is not supportive enough. Since it is an 'old' repo, you might have to try the 'old' settings like 1080Ti and CUDA8 where the environment is fully tested.
Don't know if it helps or not, but I created conda env with:
conda install scipy==1.0.0 (must be installed first) conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
and Net-Dissection-Lite worked perfect (on linux, never successed on windows).
@cyizhuo that worked perfectly, thank you so much!
@cyizhuo that worked perfectly, thank you so much!
Glad to know that~
It seems the repo is suffering from changes in Torch source code, more specifically in the use of the BatchNorm2d object put possibly on more places.
The repo works fine for the resnet18 model, but downloading and using the pretrained models for resnet50 and densenet161 yields errors:
AttributeError: 'BatchNorm2d' object has no attribute 'track_running_stats'
There are also a set of SourceChangeWarnings, among other:
Tried running both Pytorch versions 0.5.0 and 1.0.0. This thread suggests downgrading Pytorch to 0.3 would make it work, will update if I get the possibility to do so.
Error Log