tensorflow.python.framework.errors_impl.UnknownError:

tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[node resnet/conv2d/Conv2D (defined at D:/myprojects/vision/mnist_cnn.py:51) ]] [Op:__inference_distributed_function_1751]

 

위와 같은 에러가 발생했다.

---

예상컨데, tensorflow-gpu에서 기본적으로 gpu의 메모리를 전부 할당하도록 설정되어있는 데, 다른 프로그램에서 메모리를 사용 중이기 때문에 발생하는 오류라고 생각해서

 

gpus = tensorflow.config.experimental.list_physical_devices('GPU')
# 텐서플로가 첫 번째 GPU에 1GB 메모리만 할당하도록 제한
tensorflow.config.experimental.set_virtual_device_configuration(gpus[0],
			[tensorflow.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])

 

코드 첫 부분에 gpu의 메모리를 1기가만 할당하도록 제한해보았더니, 문제 없이 다시 코드가 돌아간다. 

 

---