인공능지 & 머선러닝/Tensorflow TensorflowLite1 Failed to get convolution algorithm. This is probably because cuDNN failed to initialize. Tensorflow-gpu 2.1.0 버전을 설치해서 사용하려고 하는데 Failed to get convolution algorithm. This is probably because cuDNN failed to initialize. 라고 뜨면서 코드를 실행할 수 없었음. Tensorflow는 Pytorch와 달리 CUDA와 cuDNN 버전에 매우 민감하다. 그렇기 때문에 이를 관리 해주어야 하는데, 버전이 달라질 때 마다 위 둘을 다시 설치하는 작업은 매우 번거로운 일이다. 그래서 이러한 관리를 편하게 하는데는 두 가지 방법이 있다. 1. Docker 사용 Tensorflow에서 공식적으로 각 버전에 대한 컨테이너를 배포하므로 이를 다운 받아 사용하면 가장 속이 편하다.. 2021. 8. 17. 이전 1 다음