docker-compose file to run Rviz inside a Docker container

docker-compose-rviz.yml

version: "3.4"

services:
  ros-rviz:
    build:
        context: .
        dockerfile: Dockerfile
    image: rviz_container:melodic
    container_name: rviz_test
    network_mode: "host"
    stdin_open: true # docker run -i
    tty: true        # docker run -t
    privileged: true
    runtime: nvidia
    environment:
      - DISPLAY=$DISPLAY
      - QT_X11_NO_MITSHM=1
      - XAUTHORITY=/tmp/.docker.xauth
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=all
    volumes:
      - /tmp/.X11-unix:/tmp/.X11-unix:rw
      - /tmp/.docker.xauth:/tmp/.docker.xauth:rw

Make sure you have apt-get installed the nvidia-docker2 package.

Also, what's the output when you execute nvidia-smi?