엔지니어스 - Engineeus

[Windows] Tensorflow GPU 문제 본문

Autonomous Tech./Error (Windows & Ubuntu)

[Windows] Tensorflow GPU 문제

Engineeus 2019. 12. 30. 01:49
728x90

 

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부

이번 포스팅에서는 Keras와 Tensorflow에서 GPU를 더 똑똑하게 사용하는 방법에 대해 알아보자. 케라스 (와 당연히 텐서플로우)를 사용한다면, GPU도 높은 확률로 사용 중일 것 이다. 근데 이놈의 텐서플로우는 d..

datamasters.co.kr

 

0 Comments
댓글쓰기 폼