Skip to content

Commit dd390bd

Browse files
authored
Update README.md
1 parent 2091f9d commit dd390bd

1 file changed

Lines changed: 10 additions & 3 deletions

File tree

README.md

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,15 +57,22 @@ If you don't need to interface with [IceTray](https://github.com/icecube/icetray
5757
```bash
5858
$ git clone git@github.com:<your-username>/graphnet.git
5959
$ cd graphnet
60-
$ conda create --name graphnet python=3.8 gcc_linux-64 gxx_linux-64 libgcc -y # Optional
60+
$ conda create --name graphnet python=3.8 gcc_linux-64 gxx_linux-64 libgcc cudatoolkit=11.5 -c conda-forge -y # Optional
6161
$ conda activate graphnet # Optional
6262
(graphnet) $ pip install -r requirements/torch_cpu.txt -e .[develop,torch] # CPU-only torch
6363
(graphnet) $ pip install -r requirements/torch_gpu.txt -e .[develop,torch] # GPU support
6464
```
6565
This should allow you to e.g. run the scripts in [examples/](./examples/) out of the box.
6666

67-
A stand-alone installation requires specifying a supported python version (see above) and ensuring that the C++ compilers (gcc) are up to date. Here, we have installed recent C++ compilers using conda (`gcc_linux-64 gxx_linux-64 libgcc`), but if your system already have recent versions (`$gcc --version` should be > 5, at least) you should be able to omit these from the setup.
68-
If you have an older system version of GCC than this installed and you install newer compiler though the above command, you should add `export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/anaconda3/lib/` to your `.bashrc` script or similar.
67+
A stand-alone installation requires specifying a supported python version (see above), ensuring that the C++ compilers (gcc) are up to date, and possible installing the CUDA Toolkit. Here, we have installed recent C++ compilers using conda (`gcc_linux-64 gxx_linux-64 libgcc`), but if your system already have recent versions (`$gcc --version` should be > 5, at least) you should be able to omit these from the setup.
68+
If you install the CUDA Toolkit and/or newer compilers the though the above command, you should add **one of**:
69+
```bash
70+
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/anaconda3/lib/
71+
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/miniconda3/lib/
72+
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/anaconda3/envs/graphnet/lib/
73+
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/miniconda3/envs/graphnet/lib/
74+
```
75+
depending on your setup to your `.bashrc` script or similar to make sure that the corresponding library files are accessible. Check which one of the above path contains the `.so`-files your looking to use, and add that path
6976

7077
</blockquote>
7178
</details>

0 commit comments

Comments
 (0)