From b67dcea6a2b4b917a39a47ea3f4672035c9557c1 Mon Sep 17 00:00:00 2001 From: Chaoming Wang Date: Sun, 3 Mar 2024 15:27:10 +0800 Subject: [PATCH 1/2] update --- brainpylib-changelog.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/brainpylib-changelog.md b/brainpylib-changelog.md index 888a9c682..bce0ac138 100644 --- a/brainpylib-changelog.md +++ b/brainpylib-changelog.md @@ -2,18 +2,20 @@ ## Version 0.3.0 +- Support `brainpy>=2.5.0` - Fix bugs on windows platform - remove all customized C++ and CUDA operators - ## Version 0.2.8 +- Support `brainpy>=2.5.0` - Fix bugs that the DLL cannot be loaded correctly when windows does not have a c++ environment, ## ~~Version 0.2.7(YANKED)~~ ## Version 0.2.6 +- Support `brainpy>=2.5.0` - Fix bugs of taichi call function for single result ## Version 0.2.5 From 0e866f44f1ab1a4566ed55b02d2b37b559667dea Mon Sep 17 00:00:00 2001 From: Chaoming Wang Date: Tue, 5 Mar 2024 19:12:12 +0800 Subject: [PATCH 2/2] update installation docs --- docs/index.rst | 14 +++++++++++--- docs/quickstart/installation.rst | 18 ++++++++++++++++-- 2 files changed, 27 insertions(+), 5 deletions(-) diff --git a/docs/index.rst b/docs/index.rst index ada4a8732..00271b41c 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -19,16 +19,24 @@ Installation pip install -U brainpy[cpu] # windows, linux, macos - .. tab-item:: GPU (CUDA) + .. tab-item:: GPU (CUDA 11.0) .. code-block:: bash - # for CUDA 11.0, Linux only pip install -U brainpy[cuda11] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html - # for CUDA 12.0, Linux only + .. tab-item:: GPU (CUDA 12.0) + + .. code-block:: bash + pip install -U brainpy[cuda12] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html + .. tab-item:: TPU + + .. code-block:: bash + + pip install -U brainpy[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html + For more information, please see `installation `_ section. diff --git a/docs/quickstart/installation.rst b/docs/quickstart/installation.rst index 6f51bfbd2..6931a1e3d 100644 --- a/docs/quickstart/installation.rst +++ b/docs/quickstart/installation.rst @@ -17,8 +17,7 @@ To install brainpy with minimum requirements (has installed ``jax`` and ``jaxlib .. code-block:: bash - pip install brainpy # for CPU - + pip install brainpy Minimum requirements (with dependencies) @@ -35,6 +34,9 @@ To install brainpy with minimum requirements (only depends on ``jax``), you can pip install brainpy[cuda11_mini] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html # for CUDA 11.0 pip install brainpy[cuda12_mini] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html # for CUDA 12.0 + # or + + pip install brainpy[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html # for google TPU CPU with all dependencies @@ -61,6 +63,18 @@ To install a GPU-only version of BrainPy, you can run +TPU with all dependencies +------------------------- + +BrainPy supports Google Cloud TPU. To install BrainPy along with appropriate versions of jax, +you can run the following in your cloud TPU VM: + +.. code-block:: bash + + pip install brainpy[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html # for google TPU + + + ``brainpylib`` --------------