sparseml Adding git clone to Dockerfile & details about packages

Adding git clone to Dockerfile & details about packages

Open corey-nm opened this issue 3 years ago • 2 comments

Following commands were tested, meaning all commands in README are tested:

docker build --build-arg CUDA_VERSION=11.1 -t sparseml_docker .
docker build --build-arg CUDA_VERSION=11.1 --build-arg GIT_CHECKOUT=main -t sparseml_docker .
docker build --build-arg CUDA_VERSION=11.1 --build-arg GIT_CHECKOUT=v1.0.1 -t sparseml_docker .
docker pull ghcr.io/neuralmagic/sparseml:1.0.1-ubuntu18.04-cu11.1
docker run -it sparseml_docker

Additionally, I tested that subsequent executions of the same command used the latest image checkpoint that they could.

Jul 21 '22 21:07 corey-nm

@bfineran @kylesayrs assigned for review

Jul 21 '22 21:07 github-actions[bot]

If this looks good I'll do pretty much the same thing in deepsparse

Jul 21 '22 21:07 corey-nm