tensorflow-gpu 에러 (해결)

CUDA, visual studio, cuDNN, pytorch 를 깔았던 어제..

(1) torch-gpu 설치
(2) CUDA 에러발생(재설치하기)
(3) CUDA 에러해결

torch-gpu는 정상적으로 설치됐는데 tensorflow-gpu는 오류가 발생했다. 이유가 무엇인지 살펴보쟈..

tensorflow-gpu 설치 조건

내 PC환경 :
NVIDIA GPU 드라이버 460.89
CUDA 11.1
cuDNN 8.0.4

---

에러 원인 : 
W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cusolver64_10.dll'; dlerror: cusolver64_10.dll not found

cuda/v11.1/lib 경로에 cusolver64_10.dll 이 없기 때문에..

진짜 없나? 찾아봤더니 cusolver64_10.dll 은 없고, 대신 cusolver64_11.dll 이 있는걸 발견했다. 

해결 방법 : cusolver64_11.dll 복사해서 이름을 cusolver64_10.dll 로 바꿔주었다. 

---

tensorflow-gpu 테스트 코드:

tf.test.is_gpu_available()
> True

from tensorflow.python.client import device_lib
device_lib.list_local_devices()
>
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 10968801769114951829
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 4815519744
locality {
  bus_id: 1
  links {
  }
}
incarnation: 17244984699328290065
physical_device_desc: "device: 0, name: GeForce RTX 2060, pci bus id: 0000:01:00.0, compute capability: 7.5"
]

 

해결완료!

 

해결 방법은 아래 게시글을 참고하였다.
https://imloil.tistory.com/entry/%ED%85%90%EC%84%9C%ED%94%8C%EB%A1%9C%EC%9A%B0Tensorflow%EC%97%90%EC%84%9C-GPU%EB%A5%BC-%EC%82%AC%EC%9A%A9%ED%95%98%EC%A7%80-%EC%95%8A%EC%9D%84-%EB%95%8C

텐서플로우Tensorflow에서 GPU를 사용하지 않을 때 해결 방법