fix: adding docker-cli to the Dockerfile in order to support pytorch inspection from within a container

Author: evilsocket

Dockerfile

@@ -7,6 +7,11 @@ ADD . /app
 RUN cargo build --release
 
 FROM debian:bullseye
-RUN apt-get update && apt-get install -y libssl-dev ca-certificates
+
+RUN apt-get update && apt-get install -y libssl-dev curl ca-certificates
+
+RUN curl -fsSL https://get.docker.com | sh
+
 COPY --from=builder /app/target/release/tman /usr/bin/tman
+
 ENTRYPOINT ["/usr/bin/tman"]

README.md

@@ -22,6 +22,8 @@
 
 ## Install with Cargo
 
+This is the recommended way to install and use the tool:
+
 ```bash
 cargo install tensor-man
 ```
@@ -40,6 +42,20 @@ To build your own Docker image for the tool, run:
 docker build . -t tman  
 ```
 
+## Note about Docker
+
+If you want to inspect PyTorch models and you are using `tensor-man` inside a container, make sure to share the docker socket from the host machine with the container:
+
+```bash
+docker run -it \
+  # these paths must match
+  -v/path/to/pytorch_model.bin:/path/to/pytorch_model.bin \
+  # allow the container itself to instrument docker on the host
+  -v/var/run/docker.sock:/var/run/docker.sock \
+  # the rest of the command line
+  tman inspect /path/to/pytorch_model.bin
+```
+
 ## Build from source
 
 Alternatively you can build the project from source, in which case you'll need to have Rust and Cargo [installed on your system](https://rustup.rs/).