CUDA error

각종 에러들을 해결 해 보자

CUDA | RuntimeError: CUDA error: no kernel image is available for execution on the device

pc에 문제가 생겨서 우분투를 한번 싹 밀고(사실 한 번이 아님.. ㅠ...) 새로 Ubuntu 20.04 LST를 설치하고 아나콘다, vscode, nvidia-driver, CUDA 설치까지 마친 후에 Yolo v5 모델을 사용하려는데 아래와 같은 에러 메세지가 떴다. RuntimeError: CUDA error: no kernel image is available for execution on the device 나는 RTX3080, CUDA 11.1, CUDNN 8.0.5, pytorch 1.8.0의 환경이었으며 이전과 동일하게 yolo v5 github를 git clone 해 와서 detect.py 파일을 실행하고자 했다. 여기저기 찾아보니 nvidia-driver 문제라기 보단 pytorch ..

토오오끼
'CUDA error' 태그의 글 목록