NotImplementedError: Could not run 'torchvision::nms' 오류해결

yolov8을 gpu에 올려서 사용하려고 cuda를 설치하니까 오류가 생겼다. 

torch.cuda.is_available()는 true가 나왔는데 호환이 안되는 것 같다.

 

찾아보니 torch, torchvision이 설치한 cuda버전을 지원하지 않아서 그런거였다.

 

이미 설치한 cuda는 현재기준 가장 최신 버전인 12.3이었지만 제어판 들어가서 cuda toolkit 삭제하고 pytorch사이트에서 지원한다고 적힌 cuda11.8을 설치했다.

 

그리고 torch와 torchvision 버전이 cuda 12.1으로 설치했기에 

 

pip3 uninstall torch torchvision으로 삭제 후 pytorch에서 환경을 체크 후 나온 

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

으로 

 

torch                  2.2.0+cu118
torchaudio             2.2.0+cu118
torchvision            0.17.0+cu118로 설치했다

 

문제 해결되고 gpu에서 잘 돌아갔다.