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

这似乎是 CUDA 库的问题pytorch 2.0.x+cu118

我采用的方法是

cd /home/user/anaconda3/env/torch/lib
sudo ln -sfn libnvrtc.so.11.8.89 libnvrtc.so

 

参考：

Convolutions are broken for PyTorch-2.0 CUDA-11.8 wheel builds · Issue #97041 · pytorch/pytorch · GitHub