[Pytorch] Error: no kernel image is available for execution on the device

CV & ML/PyTorch

제이스핀 2020. 1. 15. 16:53

코드들을 돌리다 보면 make.sh를 실행시켜 레이어를 만드는 경우가 많다.

docker로 환경을 받아 왔더라도 그래픽 카드 버전과 맞지 않는 경우 아래와 같은 에러를 내는 경우가 있다.

"no kernel image is available for execution on the device..."

make.sh 파일을 열어보면

arch=compute_52, code=sm_52

위와 같은 코드를 볼 수 있는데

이를 아래 사이트에서 자신이 사용하고 있는 그래픽카드에 맞춰 수정하자.

(보통 코드 전체 검색에서 'sm_' 라고 검색하면 대부분 나온다.)

예를들어

RTX2080을 사용하고 있다면 compute_75, sm_75로,

Gtx 10808을 사용하고 있다면 compute_61, sm_61로 수정하여 사용하자.

출처

https://github.com/jwyang/faster-rcnn.pytorch/issues/369