tensorflow 1.3에는 cudnn 6.0

2017. 9. 10. 18:37

anaconda가 꼬여서 새로 설치를 했습니다. 원래 cuda toolkit과 cudnn등도 다 세팅을 해둔 상태여서 anaconda만 새로 깔고 tensorflow만 새로 깔면 될 줄 알았습니다. 헌데 오류가 나서 찾아보니 cudnn 5.1이 깔려있었는데 tensorflow 1.3이 깔리면서 생긴 문제였습니다.


요컨대, tensorflow 1.3에는 cudnn 6.0을 설치해주면 됩니다.

Deep Learning/Tensorflow