From 08e6ff015d50da48559be5a4c3478b45b1291206 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Wed, 25 Sep 2024 14:26:38 +0100 Subject: [PATCH] install pytorch-cuda & tensorflow-gpu --- README.md | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index cf913c1..8fe5fff 100644 --- a/README.md +++ b/README.md @@ -29,13 +29,10 @@ Additional dependencies may be specified via `apt.txt`, `environment.yml`, and/o ```yml name: winning-submission - channels: [conda-forge, pytorch, nvidia] + channels: [conda-forge, nvidia] dependencies: - cupy - cuda-version =11.8 - - pytorch-cuda =11.8 - - tensorflow-gpu - - cudatoolkit =11.8.0 - pip - pip: - git+https://github.com/MyResearchGroup/prize-winning-algos @@ -45,8 +42,6 @@ Additional dependencies may be specified via `apt.txt`, `environment.yml`, and/o ```txt cupy-cuda11x - torch --index-url https://download.pytorch.org/whl/cu118 - tensorflow[and-cuda] git+https://github.com/MyResearchGroup/prize-winning-algos ``` @@ -68,11 +63,16 @@ The organisers will execute (after installing [nvidia-docker](https://docs.nvidi 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 ```