Skip to content

Enabling Nvidia Support in Docker

Kaiyu Zheng edited this page Jul 28, 2022 · 3 revisions

Step 1: Install NVidia Run Time

Install nvidia-docker. That will provide you with the ability to use the tag --runtime=nvidia for docker run. Follow the installation guide. It is just a few steps. Make sure their test works.

Step 2: Build Docker Image with NVidia Support

Go to robotdev/docker, then run:

source build.noetic.sh --nvidia --tag-suffix=nvidia

This will create a docker image with tag "robotdev:noetic-nvidia"

Step 3: Run a Container

Start a container from the image created above. Again, be at robotdev/docker, and then run:

source run.noetic.sh --gui --nvidia --tag-suffix=nvidia