Task: Fix TensorFlow GPU Memory Allocation for MNIST Neural Network

Task: Fix TensorFlow GPU Memory Allocation for MNIST Neural Network: a task in Terminal-Lego-15k (Harbor dataset). When running a TensorFlow-GPU MNIST neural network, the program crashes with a CUBLAS STATUS ALLOC FAILED error. This occurs because TensorFlow by default attempts to allocate all…