인프런 커뮤니티 질문&답변
작성한 질문수
error!!
작성
·
104
0
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above 에러가 났습니다!. GPU사용에 대해서 문제가 생긴거 같은데 해결 방법이 궁금합니다!
답변 1
0
AISchool
지식공유자
안녕하세요~. 반갑습니다.
1. cuda, cudnn 정상설치 유무확인
tensorflow gpu버전을 사용하기 위해서는 cuda와 cudnn이 먼저 설치되어야만 합니다. 설치 방법은 아래 링크를 참조하세요~.
https://www.tensorflow.org/install/gpu?hl=ko
2. cuda, cudnn이 정상 설치되었을경우
cuda와 cudnn이 정상 설치되어도 해당 에러가 발생할경우 gpu 맵핑 문제일 수있습니다. gpu 맵핑 문제를 해결하기 위해 tf import 후에 아래 코드를 추가해서 다시 실행해보세요. (참조 : https://www.tensorflow.org/guide/gpu?hl=ko)
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
# 텐서플로가 첫 번째 GPU만 사용하도록 제한
try:
tf.config.experimental.set_visible_devices(gpus[0], 'GPU')
except RuntimeError as e:
# 프로그램 시작시에 접근 가능한 장치가 설정되어야만 합니다
print(e)
감사합니다~.