/docker-spacy-gpu

Minimal example of a GPU Docker container that runs SpaCy Transformers

Primary LanguageDockerfileMIT LicenseMIT

Spacy Transformers with GPU support in Docker

Minimal image for GPU Docker container that runs SpaCy Transformers. To use docker-compose you need nvidia-docker runtime.

The models are stored in a /model volume outside the container, so make sure that downloads first with make model.

First, enable Docker Experimental. To build all the containers:

make

To run the test:

make run

or

docker-compose up

or

nvidia-docker run --name spacygpu --rm -it -v ${PWD}/model:/model spacygpu:latest

or

docker run --name spacygpu --runtime=nvidia --rm -it --gpus all -v ${PWD}/model:/model spacygpu:latest