tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
WLS2002
2024-07-13 21:28:53 +08:00
committed by GitHub
parent 2b391a9b33
commit fda28a35b6
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
README.md
View File

@@ -37,7 +37,22 @@ TensorNEAT is a JAX-based libaray for NeuroEvolution of Augmenting Topologies (N
- Test neuroevolution algorithms on advanced **RL tasks** (Brax, Gymnax). - Test neuroevolution algorithms on advanced **RL tasks** (Brax, Gymnax).
## Installation ## Installation
Install `tensorneat` from the GitHub source code:
1. Install the correct version of [JAX](https://github.com/google/jax). We recommend `jax >= 0.4.28`.
For cpu version only, you may use:
```
pip install -U jax
```
For nvidia gpus, you may use:
```
pip install -U "jax[cuda12]"
```
For details of installing jax, please check https://github.com/google/jax.
2. Install `tensorneat` from the GitHub source code:
``` ```
pip install git+https://github.com/EMI-Group/tensorneat.git pip install git+https://github.com/EMI-Group/tensorneat.git
``` ```