This is the first post of devmuz
잘 되던 코드가 갑자기 에러가 막!막! 떠서 뭘 구글링 할지도 모르겠었는데 운 좋게 찾았다. 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
Comments
Post a Comment