Run pytorch on GPU with CUDA

Question

I tried to run all the computations on gpu, but the increase in the speed wasn\'t as big as i expected, also nvidia-smi gives this: nvidia-smi output, while pro