Run in Linux error: RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM
wangwyForNewTime opened this issue · 1 comments
wangwyForNewTime commented
run predict.py in Windows successfully
but run in Linux ，the error happened :RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM
What I do in Linux is the same as in Windows.
Used pretrained model:fpn_inception.h5
How I can do to change this?Thank you
wangwyForNewTime commented
I find the solution：use newer pytorch version
I run in Windows on my laptop computer with low configuration，so, I use :conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=9.0 -c pytorch
I run in Linux on the studio's computer with 2080ti.When I used pytorch==1.0.1, the error occured.
I try conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=9.2 -c pytorch,and the coding run successfully