Hi , Would you release your experiments environment?
Johnson-yue opened this issue · comments
I am using, python3.7.3, cuda 10.0, cudnn 7.6.1 and then I run finetune.py as command:
python finetune.py --name AnimalFace_finetune --mixing --loss r1 --sched --dataset AnimalFace
and the Bug is :
cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input
bug occur in line 312 in finetune.py
Do you understand this bug meaning???
Sorry this is environment bug, and I fixed with:
torch : 1.2.0
torchvision : 0.3.0
cuda 10
Hi, sorry for late response! Glad to hear that you fixed it! :)
Ran into the same issue. Managed to get it working with a nightly pytorch build:
torch: 1.5.0
torchvision: 0.6.0
cuda 9.1
@LendelTheGreat Thank you
Hi, sorry for the late response!
I remember that I had the same issue, but I forgot it since I fixed it a while ago.
It is a bug in PyTorch 1.4.0 (when one using .transpose() or .permute()), and one can fix it by upgrading to nightly version or downgrading. Thanks for the reminder @LendelTheGreat !