Language/Python

[Tensorflow] import error, 텐서플로우 임폴트시 발생하는 에러

쩡 (개발꿈나무) 2021. 9. 25. 09:16

2021-09-25 09:10:00.736688: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found                              2021-09-25 09:10:00.737110: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

 

cuda 10.1 에서 pip install tensorflow를 했을 경우, gpu 사용이 안 됨.

cuda에서 사용하는 (?) 텐서플로우 버전이 다르게 깔리기 때문.

이 경우 cuda 또는 cudnn 을 업그레이드 하거나, pip install tensorflow==[버전] 으로 설치해주면 사용 가능.

'Language > Python' 카테고리의 다른 글

[Anaconda] 아나콘다 설치 및 세팅  (0) 2021.07.14
[Python 3.8.7] Windows10에 파이썬 설치  (0) 2021.01.21