wangchunlin
/
cvat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
main
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '6cb05c8a8f'
${ noResults }
cvat/components/cuda/README.md

1.1 KiB
Raw Blame History

NVIDIA CUDA Toolkit

Requirements

  • NVIDIA GPU with a compute capability [3.0 - 7.2]
  • Latest GPU driver

Installation

Install the latest driver for your graphics card

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-cache search nvidia-*   # find latest nvidia driver
sudo apt-get install nvidia-*    # install the nvidia driver
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
sudo apt-get install nvidia-modprobe

Reboot your PC and verify installation by nvidia-smi command.

Install Nvidia-Docker

Please be sure that installation was successful.

docker info | grep 'Runtimes'   # output should contains 'nvidia'

Build docker image

# From project root directory
docker-compose -f docker-compose.yml -f components/cuda/docker-compose.cuda.yml build

Run docker container

# From project root directory
docker-compose -f docker-compose.yml -f components/cuda/docker-compose.cuda.yml up -d