-
(tensorflow) tensorflow 2.2.0와 호환되는 cuda, cuDNN 버전IT 지식 창고 2020. 9. 10. 18:15
현재 Test를 해본 결과
tensorflow = 2.2.0
cuda = 10.1
cuDNN = 7.6.5 for cuda 10.1
위와 같이 설치해야 에러가 안납니다.
만약 위와 같이 설치 안할 경우 나타나는 에러는
1. cudart64 101.dll not found
-> 이것은 cuda 10.1 버전을 설치해야 생기는 .dll 파일 입니다. 뒤에 101이 버전을 따라가는 것으로 11버전을 설치하면 cudart64 110.dll 파일이 설치 됩니다.
2. cudnn64_7.dll not found
-> 이것은 cuDNN을 7.6.5버전으로 설치해야 합니다. 간혹 for cuda 10.1만 보고 8.0.x버전을 설치하는 경우가 있는데, 이 경우에도 cudnn64_8.dll로 설치되기 때문에 7.6.5버전에 cuda버전과 맞춰 설치하시면 됩니다.
아직 tensorflow 2.3.0으로는 테스트를 안해봤는데, 현재 2.2.0으로 진행하는 프로젝트를 2.3.0으로 업그레이드 시켰을 때 발생하는 문제를 해결하기가 두려워 아직 테스트를 안해봤습니다...
'IT 지식 창고' 카테고리의 다른 글
(keras) model과 weight 종속성 (0) 2020.09.22 (Pillow) alpha값을 이용한 이미지 합성 (0) 2020.09.17 (tensorflow) tensorflow 2.x GPU 메모리 제한 후 필요시 증가 (0) 2020.09.10 (albumentation) pytorch가 없다고 뜰 때 (3) 2020.09.08 Google Colaboratory Auto Drive Mount (자동 드라이브 마운트 ) (0) 2020.09.07 댓글