Detectron2 CUDNN_STATUS_NOT_SUPPORTED error when training a model in WSL. #5284
-
I'm getting this error when training a model in detectron2 on my PC in WSL2 UBUNTU 22.04. I don't get this error in Colab. I have installed Cuda 12.2 and cuDNN 8.9.6. |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
Uninstalled PyTorch 2.3.0 and installed 2.2.2 and trained model without error. |
Beta Was this translation helpful? Give feedback.
Uninstalled PyTorch 2.3.0 and installed 2.2.2 and trained model without error.
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia
See: pytorch/pytorch#121834