Tensorflow GPU version - InternalError: Blas GEMV launch failed
잘 되던 코드가 갑자기 에러가 막!막! 떠서 뭘 구글링 할지도 모르겠었는데 운 좋게 찾았다.
1
2
3
4
|
# Assume that you have 12GB of GPU memory and want to allocate ~4GB:
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
| cs |
윈도우에서 Tensorflow GPU 버전을 이용할 때 일어나는 문제라는데,
세션을 그냥 tf.Session()으로 열지말고 위의 코드 대로 열면 해결 되더라.
출처 - https://stackoverflow.com/questions/34199233/how-to-prevent-tensorflow-from-allocating-the-totality-of-a-gpu-memory/34200194
0 comments