[Windows] Tensorflow GPU 문제

 

Tensorflow 2에서 자꾸 아래와 같은 오류가 나서 고생 했습니다.

2019-12-30 01:42:43.383561: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cublas64_100.dll'; dlerror: cublas64_100.dll not found 
2019-12-30 01:42:43.383761: E tensorflow/stream_executor/cuda/cuda_blas.cc:238] failed to create cublas handle: CUBLAS_STATUS_INTERNAL_ERROR 
2019-12-30 01:42:43.384698: E tensorflow/stream_executor/cuda/cuda_blas.cc:238] failed to create cublas handle: CUBLAS_STATUS_INTERNAL_ERROR 
2019-12-30 01:42:43.385885: E tensorflow/stream_executor/cuda/cuda_blas.cc:238] failed to create cublas handle: CUBLAS_STATUS_INTERNAL_ERROR 
2019-12-30 01:42:43.386065: W tensorflow/stream_executor/stream.cc:1919] attempting to perform BLAS operation using StreamExecutor without BLAS support 
2019-12-30 01:42:43.386224: W tensorflow/core/common_runtime/base_collective_executor.cc:216] BaseCollectiveExecutor::StartAbort Internal: Blas GEMV launch failed:  m=1, n=4 
 [[{{node sequential/dense/MatMul}}]]





[Op:__inference_distributed_function_418]



Function call stack: 
distributed_function

 

찾아보니 GPU할당 문제라서 NVIDIA GPU인 'GPU 1'만 지정하도록 아래와 같이 강제로 해버렸습니다.

 

os.environ["CUDA_VISIBLE_DEVICES"] = '1'

 

---

Reference

https://datamasters.co.kr/33

Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 1부