File tree 1 file changed +6
-1
lines changed 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -42,12 +42,17 @@ You can try running Graphium on Graphcore IPUs for free on Gradient by clicking
42
42
43
43
### For CPU and GPU developers
44
44
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 ` .
46
48
47
49
``` bash
48
50
# Install Graphium's dependencies in a new environment named `graphium`
49
51
mamba env create -f env.yml -n graphium
50
52
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
+
51
56
# Install Graphium in dev mode
52
57
mamba activate graphium
53
58
pip install --no-deps -e .
You can’t perform that action at this time.
0 commit comments