From f31a40bf38526b100a5f2216ab66b8d5466114ee Mon Sep 17 00:00:00 2001 From: DomInvivo <47570400+DomInvivo@users.noreply.github.com> Date: Thu, 30 Nov 2023 11:53:23 -0500 Subject: [PATCH] Update README.md --- README.md | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index a83f7ab40..bc5a1aea4 100644 --- a/README.md +++ b/README.md @@ -42,12 +42,17 @@ You can try running Graphium on Graphcore IPUs for free on Gradient by clicking ### For CPU and GPU developers -Use [`mamba`](https://github.com/mamba-org/mamba): +Use [`mamba`](https://github.com/mamba-org/mamba), a faster and better alternative to `conda`. + +If you are using a GPU, we recommend enforcing the CUDA version that you need with `CONDA_OVERRIDE_CUDA=XX.X`. ```bash # Install Graphium's dependencies in a new environment named `graphium` mamba env create -f env.yml -n graphium +# To force the CUDA version to 11.2, or any other version you prefer, use the following command: +# CONDA_OVERRIDE_CUDA=11.2 mamba env create -f env.yml -n graphium + # Install Graphium in dev mode mamba activate graphium pip install --no-deps -e .