Skip to content

Latest commit

 

History

History
111 lines (79 loc) · 2.88 KB

README.md

File metadata and controls

111 lines (79 loc) · 2.88 KB

ASOne

Table of Contents

Docker Installation

Ubuntu

Prerequisite

  1. For systems with GPU please verify you have nvidia drivers installed. run
nvidia-smi

if you see something like the following. Then you can continue running shell script

or you can install nvidia drivers using following command:

sudo apt-get install nvidia-driver-YYY nvidia-dkms-YYY
  • YYY= Nvidia driver version

e.g sudo apt-get install nvidia-driver-510 nvidia-dkms-510

  • Reboot your system after installing nvidia-drivers.
sudo reboot

Install using Shell Script

chmod a+x docker-installation.sh
./docker-installation.sh
  • For systems with GPU run following commands after installing docker.

Setup the package repository and the GPG key:

 distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
    && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
    && curl -s -L https://nvidia.github.io/libnvidia-container/experimental/$distribution/libnvidia-container.list | \
       sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
       sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

Install the nvidia-docker2 package (and dependencies) after updating the package listing:

sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

sudo xhost +local:docker
  • [NOTE] If there is an error while installing docker, try removing apt-lists and resinstalling.
sudo rm -rf /var/lib/apt/lists/*
./docker-installation.sh
  • In case shell script keeps failing or you want to install manually follow steps in Manual Installation

Test Docker

Test Using Docker Compose File

  1. Run container without GPU
docker compose run linux
  1. Run container with GPU
docker compose run linux-gpu
  • To test DISPLAY is shared with docker properly:
cd asone-linux
python main.py

  • if an image show up then everything is working properly.

  • if you see an error saying qt.qpa.xcb: could not connect to display that means your display is not accessible to docker.

Try this:

sudo xhost +local:docker
  • To build and run docker container manually follow instructions for Manual Build

Test Detectron2

To test detectron2 follow instuctions in Demo Detectron2