[Python] / [Error] Could not load dynamic library 'libcusolver.so.10' - tensorflow 에러
·
Python/Error
문제점 Pytcharm에서 tensorflow 프레임워크에서 모델을 학습시키려고하는데 학습은 진행되고 있지만, CPU를 사용하여 학습 속도가 매우 느린편이였다. 즉, GPU를 전혀 사용하지 않는것이다. nvidia-smi 사용가능한 device 목록을 다음코드로 찾아보았다. from tensorflow.python.client import device_lib device_lib.list_local_devices() 결과, CPU만 목록에 나타나고 GPU는 잡지 못하고 있었다. 나의 생각으로는 Pytorch에서는 GPU를 사용하여 학습이 잘 되었기 때문에 CUDA 버전 문제는 아니라고 생각했다. 또한, CPU지만 학습을 할 수 있었기 때문에, tensorflow, kears 등 Library나 Python..
욱근욱
'Python/Error' 카테고리의 글 목록