Hello! Please add video memory usage! `nvidia-smi` provides info. For coding with Pytorch and Tensorflow using CUDA we need this info! Thanks very much.