Add CUDA_VISIBLE_DEVICES environment variable when using the --gpus flag#345

sergey-serebryakov merged 8 commits into

mlcommons:mastermlcommons/mlcube:masterfrom

davidjurado:feature/docker_gpudavidjurado/mlcube:feature/docker_gpuCopy head branch name to clipboard