UserWarning: Applied workaround for CuDNN issue, install nvrtc.so

[arr0i3]

Dear all,

I have encountered this UserWarning when running model_angelo build_no_seq:

~/anaconda3/envs/model-angelo_local/lib/python3.10/site-packages/torch/nn/modules/conv.py:605: UserWarning: Applied workaround for CuDNN issue, install nvrtc.so (Triggered internally at /opt/conda/conda-bld/pytorch_1711403388920/work/aten/src/ATen/native/cudnn/Conv_v8.cpp:80.)

Following this StackOverflow question I have added a symbolic link at ~/anaconda3/envs/model-angelo_local/lib:

libnvrtc.so -> libnvrtc.so.11.8.89

besides the one that was generated automatically:

libnvrtc.so.11.2 -> libnvrtc.so.11.8.89

By running ln -sfn libnvrtc.so.11.8.89 libnvrtc.so

And now there is no warning.
Hope it helps :)

[victorkitov]

It helped, thank you!

[jamaliki]

Thank you @arr0i3 , I am not sure why this error occurs. I will pin your issue.