[Issue] failed call to cuInit: CUDA_ERROR_UNKNOWN: unknown error

ubuntu 20.04 lts 환경에서 cuda, cudnn을 설치하고 난 후 텐서플로우에서 gpu를 사용할 수 있는지 확인하고자 했습니다.

 

확인하려고 한 명령어는 다음과 같습니다.

 

import tensorflow as tf
tf.config.list_physical_devices('gpu')

 

그랬더니 에러가 발생했..

 

에러 내용

failed call to cuInit: CUDA_ERROR_UNKNOWN: unknown error

 

검색 결과..

해결방안

sudo apt install nvidia-modprobe

 

 

 

cuda, cudnn을 설치하고자 참고한 사이트 : https://webnautes.tistory.com/1428

Ubuntu 20.04에 CUDA Toolkit와 cuDNN, Tensorflow 설치하기