Skip to content

Commit f31a40b

Browse files
authored
Update README.md
1 parent a8e715c commit f31a40b

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

README.md

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,17 @@ You can try running Graphium on Graphcore IPUs for free on Gradient by clicking
4242

4343
### For CPU and GPU developers
4444

45-
Use [`mamba`](https://github.com/mamba-org/mamba):
45+
Use [`mamba`](https://github.com/mamba-org/mamba), a faster and better alternative to `conda`.
46+
47+
If you are using a GPU, we recommend enforcing the CUDA version that you need with `CONDA_OVERRIDE_CUDA=XX.X`.
4648

4749
```bash
4850
# Install Graphium's dependencies in a new environment named `graphium`
4951
mamba env create -f env.yml -n graphium
5052

53+
# To force the CUDA version to 11.2, or any other version you prefer, use the following command:
54+
# CONDA_OVERRIDE_CUDA=11.2 mamba env create -f env.yml -n graphium
55+
5156
# Install Graphium in dev mode
5257
mamba activate graphium
5358
pip install --no-deps -e .

0 commit comments

Comments
 (0)