Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
- Buildkit should be default by now
- Dockerfile now supports both GPU and CPU
  • Loading branch information
gb-beng authored Jul 26, 2024
1 parent 8547b35 commit 16f3867
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,11 @@ Apply VisXP models trained for feature extraction from keyframes and audio spect

From the root of the repo, run
```
DOCKER_BUILDKIT=1 docker build -t dane-visual-feature-extraction-worker .
docker build -t dane-visual-feature-extraction-worker .
```
Buildkit is optional, it may speed up building (see https://docs.docker.com/build/buildkit/)
NB: building the image has occasionally led to mysterious connection errors, which we haven't been able to track down and solve (yet). Discarding the poetry.lock has been a way to circumvent these.

The default Dockerfile is based on an image that allows GPU processing. A Dockerfile based on a regular Python base image is provided in `Dockerfile-noGPU`, but should not be essential.
Whether the processing actually uses GPU depends on the availability of GPU in the container.
The Dockerfile support both CPU and GPU processing. Whether the processing actually uses GPU depends on the availability of GPU in the container.

### Config

Expand Down

0 comments on commit 16f3867

Please sign in to comment.