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.
a5c3da1235
* Replaced wget by curl * Moved CI stuff into Dockerfile.ci * Use docker-compose to run commnands inside docker (need environment variables) * Added patool again (to support different archive formats) * Roll back tensorflow version: 1.15 -> 1.13.1 Fixed https://github.com/opencv/cvat/issues/982 Fixed https://github.com/opencv/cvat/issues/1017 * datumaro install tensorflow 2.x now. It breaks automatic annotation using TF. * Follow redirects in curl (auto_segmentation) |
6 years ago | |
|---|---|---|
| .. | ||
| README.md | 7 years ago | |
| docker-compose.cuda.yml | 7 years ago | |
| install.sh | 6 years ago | |
README.md
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