Skip to content

Commit

Permalink
updated.
Browse files Browse the repository at this point in the history
  • Loading branch information
tanyuqian committed Dec 7, 2023
1 parent bace993 commit 469bed2
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 10 deletions.
10 changes: 1 addition & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,17 +17,9 @@ Here is also a [Quick Tutorial](tutorials/quick.md) for you to become an expert
```
pip install redco
```
For the most up-to-date version:
```
git clone https://github.com/tanyuqian/redco.git
cd redco
pip install -e .
```

#### Adjust Jax & Flax versions
The command above would automatically install cpu version of jax, so the version of Jax need to be adjusted based on your device.

For example,
The command above would automatically install cpu version of jax, so the version of Jax need to be adjusted based on your device. For example,
```
pip install --upgrade flax==0.7.0
pip install --upgrade jax[cuda11_pip]==0.4.13 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

setup(
name="redco",
version="0.4.14dev",
version="0.4.14",
author="Bowen Tan",
packages=find_packages(),
install_requires=['jax', 'flax', 'optax', 'numpy'],
Expand Down

0 comments on commit 469bed2

Please sign in to comment.