I am implementing the following project: https://github.com/sinhasam/vaal 1
and it is not using the system’s GPU. Kindly check main.py

PyTorch Version : 1.1.0
OS: Ubuntu
How you installed PyTorch: pip
Python version: 3
CUDA/cuDNN version:
GPU models and configuration:
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61