Skip to content

Commit

Permalink
Remove unnecessary dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
Routhleck committed Dec 4, 2024
1 parent 589a27c commit a48e3b8
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 11 deletions.
8 changes: 1 addition & 7 deletions docs/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -20,13 +20,7 @@ Installation
# python 3.9-3.11
pip install -U brainpy[cpu] # windows, linux, macos
.. tab-item:: GPU (CUDA 11.0)

.. code-block:: bash
pip install -U brainpy[cuda11] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
.. tab-item:: GPU (CUDA 12.0)
.. tab-item:: GPU (CUDA 12)

.. code-block:: bash
Expand Down
6 changes: 2 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,12 +68,10 @@
'https://storage.googleapis.com/jax-releases/jax_cuda_releases.html',
],
extras_require={
'cpu': ['jaxlib>=0.4.13', 'brainpylib', 'numba', 'braintaichi'],
'cuda11': ['jaxlib[cuda11_pip]', 'brainpylib', 'numba', 'braintaichi'],
'cuda12': ['jaxlib[cuda12_pip]', 'brainpylib', 'numba', 'braintaichi'],
'cpu': ['jaxlib>=0.4.13', 'numba', 'braintaichi'],
'cuda12': ['jaxlib[cuda12_pip]', 'numba', 'braintaichi'],
'tpu': ['jaxlib[tpu]', 'numba',],
'cpu_mini': ['jaxlib>=0.4.13'],
'cuda11_mini': ['jaxlib[cuda11_pip]'],
'cuda12_mini': ['jaxlib[cuda12_pip]'],
},
keywords=('computational neuroscience, '
Expand Down

0 comments on commit a48e3b8

Please sign in to comment.