본문 바로가기

카테고리 없음

Torch GPU 사용 설정 에러: CUDA initialization: CUDA unknown error

torch GPU 사용 설정 중 조우한 에러

 

"""

CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment

"""

 

해결하기 위해선 아래 명령어를 입력해야 했다.

> sudo rmmod nvidia_uvm
> sudo modprobe nvidia_uvm

 

첫 명령어를 통해서, 내가 nvidia_uvm이라는 모듈 자체를 갖고 있지 않았다는 사실을 알았고, 두 번째 명령어를 통해 제대로 설치할 수 있었다. 이후 정상 사용가능했다.