[pytorch] RuntimeError: cuda runtime error (48) : no kernel image is available for execution on the device at

[Make로 레이어 만들고, 모델 돌렸을 때 나는 오류인 경우]

src/cuda/Makefile 파일들을 수정해서 해결 가능. (보통 roi_aling 등의 특별한 네트워크 추가 시에 makefile이 존재)

파일을 열어보면

-gencode arch=compute_61,code=sm_61

라는걸 볼 수 있을 텐데, 61 혹은 다른 숫자를 아래 사이트에 나온 자신의 그래픽카드 번호에 맞게 설정

https://developer.nvidia.com/cuda-gpus

그래픽카드 확인 명령어

lspci |grep -i VGA

만약 5.2라면 52라고 수정

ex) GeForce GTX TITAN X -> 52

-> -gencode arch=compute_52,code=sm_52

위와같은 Makefile은 여러개가 있는 경우가 많으니, 다 찾아서 바꿔줘야함

참고로 TITAN X는 Nvidia도 있고, GeForce GTX도 있으니 잘 구별해야함