새소식

TroubleShooting

[Error] Failed to get convolution algorithm. This is probably because cuDNN failed to initialize(Jupyter lab)

  • -

 

 

주피터랩에서 텐서플로우를 실행하는데 갑자기 아래와 같은 코드가 떠서 당황했다.

 

 

에러 유형(캡처)

1. 일반적인 에러
2. 다른 유형의 에러

 

 

에러 유형(텍스트) - 위 캡처와 동일합니다.

1
2
3
4
5
UnknownError:  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.
     [[node sequential/conv2d/Conv2D (defined at <ipython-input-8-c246912e5591>:1) ]] [Op:__inference_train_function_841]
 
Function call stack:
train_function
cs
1
2
3
4
5
InternalError:  Blas GEMM launch failed : a.shape=(32, 784), b.shape=(784, 128), m=32, n=128, k=784
     [[node sequential/dense/MatMul (defined at <ipython-input-9-49e902c4b07e>:1) ]] [Op:__inference_train_function_571]
 
Function call stack:
train_function
cs

 

 

 

이유를 찾아봤는데 다른 파일(.ipynb)에 텐서플로우를 실행하고 커널이 할당되어 있는 상태에서 위와 같은 오류가 난 것으로 판단, 실행하는 파일의 커널만 살리고 나머지는 셧다운시켰더니 해결되었다.(아래 참고)

내가 사용할 파일인 0414.ipynb의 커널만 사용한다.(나머지 파일은 SHUT DOWN시켰다.)

 

 

 

기타 참고할 만한 링크(이것은 멀티 GPU에 관한 해결법) : bart-2012.tistory.com/18

 

 

 

 

 

 

Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.