... | ... | @@ -22,3 +22,24 @@ To update: |
|
|
$ git pull
|
|
|
$ docker-compose up -d --build
|
|
|
|
|
|
|
|
|
## Use your GPU:
|
|
|
1) Install cuda & cudnn
|
|
|
2) Create a file named `local.yml` with this content
|
|
|
```
|
|
|
version: "2.4"
|
|
|
|
|
|
services:
|
|
|
celery-gpu:
|
|
|
shm_size: '3gb'
|
|
|
runtime: nvidia
|
|
|
environment:
|
|
|
- KRAKEN_TRAINING_DEVICE=cuda:0
|
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
|
|
- NVIDIA_DRIVER_CAPABILITIES=all
|
|
|
|
|
|
```
|
|
|
3) When you build use both yml files
|
|
|
```
|
|
|
$ docker-compose -f docker-compose.yml -f local.yml up -d --build
|
|
|
``` |
|
|
\ No newline at end of file |