Enable GPU compute for controllers

[omichel]

This PR enables the GPU compute inside the docker, so that controllers using GPU (Pytorch, TensorFlow, etc.) can get hardware acceleration as well.

[BenjaminDeleze]

It seems pretty useless to me as these variables are already set by Nvidia:

The only difference I see is the compute and compat32

[omichel]

Yes, but I believe compute is the key one needed for enabling GPU compute. And if we set only ENV NVIDIA_DRIVER_CAPABILITIES compute, I am afraid it will erase graphics and utility which are also needed.

[omichel]

But yes, we can probably remove ENV NVIDIA_VISIBLE_DEVICES all since it appears to be already set.

[BenjaminDeleze]

I found the list of capabilities:

Supported driver capabilities
compute: required for CUDA and OpenCL applications.
compat32: required for running 32-bit applications.
graphics: required for running OpenGL and Vulkan applications.
utility: required for using nvidia-smi and NVML.
video: required for using the Video Codec SDK.
display: required for leveraging X11 display.

So, yes it seems that we need compute