Fix wrong docker run command

[gqqnbig]

It forgot the image name ubuntu.

- How to verify it

docker run -it --rm --gpus '"device=0,2"' nvidia-smi throws error

Unable to find image 'nvidia-smi:latest' locally
docker: Error response from daemon: pull access denied for nvidia-smi, repository does not exist or may require 'docker login': denied: requested access to the resource is denied.
See 'docker run --help'.

but mine does not.

[thaJeztah]

LGTM

did a quick rebase to get CI to run, and will merge once that finishes

[codecov-commenter]

Codecov Report

Merging #3558 (90a1887) into master (dd3ba73) will not change coverage.
The diff coverage is n/a.

Additional details and impacted files

@@           Coverage Diff           @@
##           master    #3558   +/-   ##
=======================================
  Coverage   59.64%   59.64%           
=======================================
  Files         287      287           
  Lines       24771    24771           
=======================================
  Hits        14774    14774           
  Misses       9111     9111           
  Partials      886      886