fast-reid/docker
KleinYuan df3d23bbd2 clean up 2023-01-27 14:58:35 -08:00
..
README.md clean up 2023-01-27 14:58:35 -08:00

README.md

Use the container

Build Container

cd docker/
# Build with the corresponding CUDA version

# CUDA 10
docker build -t=fastreid:v0 -f Dockerfile_CUDA10 .
# CUDA 11
docker build -t=fastreid:v0 -f Dockerfile_CUDA11 .

Run Container

# Launch (requires GPUs)
nvidia-docker run -v ${PWD}:/home/appuser --name=fastreid --net=host --ipc=host -it fastreid:v0

Run Training

Next, follow the Get Started Doc.

A more complete docker container

If you want to use a complete docker container which contains many useful tools, you can check my development environment Dockerfile