from device: CUDA_ERROR_OUT_OF_MEMORY

2019-03-06 10:15:43.999337: E C:\users\nwani\_bazel_nwani\swultrt5\execroot\org_tensorflow\tensorflow\stream_executor\cuda\cuda_driver.cc:936] failed to allocate 2.3K (2304 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY
2019-03-06 10:15:43.999337: E C:\users\nwani\_bazel_nwani\swultrt5\execroot\org_tensorflow\tensorflow\stream_executor\cuda\cuda_driver.cc:936] failed to allocate 2.3K (2304 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY
2019-03-06 10:15:43.999337: E C:\users\nwani\_bazel_nwani\swultrt5\execroot\org_tensorflow\tensorflow\stream_executor\cuda\cuda_driver.cc:936] failed to allocate 2.3K (2304 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY
2019-03-06 10:15:44.000337: E C:\users\nwani\_bazel_nwani\swultrt5\execroot\org_tensorflow\tensorflow\stream_executor\cuda\cuda_driver.cc:936] failed to allocate 2.3K (2304 bytes) from

Check the nvidia driver version, cuda version, cudnn version, python version, tensorflow version

Under windows, you can change the tensorflow version by following in cmd:
$ activate tensorflow
$ pip install tensorflow==1.4.0