shepherd's Blog

tensorflow-gpu 1.14 import error(cuda version error) 본문

python

tensorflow-gpu 1.14 import error(cuda version error)

shepherd.dev 2019. 7. 16. 19:09

tensorflow-gpu 1.14 cuda dll import error

tensorflow-gpu 1.14의 경우 현재 cuda 10.1 을 지원하지 않는 것 같습니다.

cuda 10.1 + cudnn 7.6 (for cuda 10.1) 을 사용할 경우 import error가 발생하여,

cuda 10.0 + cudnn 7.6 (for cuda 10.0) 을 사용하여 문제를 해결하였습니다.

git issue에서 tensorflow 2.0 beta를 사용해 보라고 하는데, 마음 편하게 cuda 10.0 이하를 쓰시는게 좋을 거 같습니다.(심지어 해결 안된듯)

link