RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /opt/conda/conda-bld/pytorch_1570910687230/work/aten/src/THC/THCGeneral.cpp:50

 

cuda가 인식이 안 돼서 발생하는 문제이다. 이를 확인해보기 위해서

 

터미널에서 python을 입력하고 들어가서 아래로 잘 확인해보기

import torch 
print(torch.__version__)

print(torch.cuda.current_device())
print(torch.cuda.device(0))
print(torch.cuda.device_count())
print(torch.cuda.get_device_name(0))
print(torch.cuda.is_available())
print(torch.cuda.current_device())

 

차례대로 토치 버전과 디바이스에서 사용하는 번호, 사용 가능한지 등을 확인할 수 있다.

 

난 모두 정상이었는데 코드 상에서 gpu 번호를 9번으로 default 값으로 잡고 있어서 발생한 오류였다. 난 0, 1번을 사용해서 이를 고쳐주니 해결되었다.

 

만약 위에서 확인했을 때, cuda를 사용할 수 없다 그러면 마지막으로 nvidia-smi로 확인한 다음 그래픽 드라이버를 설치하면 된다.

 

그래도 다 안 된다면

 

device = 'cuda' if torch.cuda.is_available() else 'cpu'
model.to(device)

 

gpu를 사용 못 할때 cpu로 연산 처리가 되도록 설정을 추가해주도록 하자.