From f8933859027187e198656d79d7f9d2fa6cad4665 Mon Sep 17 00:00:00 2001 From: "Sam.Porter" Date: Mon, 30 Sep 2024 19:58:01 +0100 Subject: [PATCH] script to run PETRIC with docker --- docker.sh | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/docker.sh b/docker.sh index c817c44..3c2eefc 100755 --- a/docker.sh +++ b/docker.sh @@ -1,10 +1,16 @@ +# 1. git clone & cd to your submission repository # 2. mount `.` to container `/workdir`: docker run --rm -it --gpus all -p 6006:6006 \ -v /path/to/data:/mnt/share/petric:ro \ -v .:/workdir -w /workdir synerbi/sirf:edge-gpu /bin/bash -# 3. optionally, conda/pip/apt install environment.yml/requirements.txt/apt.txt -# 4. install metrics & run your submission +# 3. install metrics & GPU libraries +conda install monai tensorboard tensorboardx jupytext cudatoolkit=11.8 +pip uninstall torch # monai installs pytorch (CPU), so remove it +pip install tensorflow[and-cuda]==2.14 # last to support cu118 +pip install torch --index-url https://download.pytorch.org/whl/cu118 pip install git+https://github.com/TomographicImaging/Hackathon-000-Stochastic-QualityMetrics +# 4. optionally, conda/pip/apt install environment.yml/requirements.txt/apt.txt +# 5. run your submission python petric.py & -# 5. optionally, serve logs at +# 6. optionally, serve logs at tensorboard --bind_all --port 6006 --logdir ./output \ No newline at end of file