cuDNN error錯誤訊息

最近跑程式, 遇到這個錯誤訊息

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

網路上看到的討論大多都是版本問題 (已確認沒問題), 或者是記憶體爆了 (有檢查RAM與GPU的使用狀態, 也沒有吃到滿)

神奇的是, 我用相同的conda環境, 在電腦A可以跑(1080TI), 電腦B(4060)就遇到這個錯誤訊息 QQ

原本的環境: (python3.7)
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

後來嘗試最新版的環境 (python 3.10)

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

結果竟然就可以了!!! 而且速度也比較快, 真神奇 XD