Skip to content

Commit 7400fb9

Browse files
authored
Update installation.rst (#80)
* Update installation.rst - simplify options, move paper repo to secondary menu * Update installation.rst - typo fixes * Update docs/source/installation.rst
1 parent eaa8940 commit 7400fb9

File tree

1 file changed

+47
-41
lines changed

1 file changed

+47
-41
lines changed

docs/source/installation.rst

Lines changed: 47 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ Installation Guide
44
System Requirements
55
-------------------
66

7-
CEBRA is written in Python (3.8+) and PyTorch. CEBRA is most effective when used with a GPU, but CPU-only support is provided. We provide instructions to run CEBRA on your system directly using Anaconda or Docker. The instructions below were tested on different compute setups with Ubuntu 18.04 or 20.04, using Nvidia GTX 2080, A4000, and V100 cards. Other setups are possible (including Windows), as long as CUDA 10.2+ support is guaranteed.
7+
CEBRA is written in Python (3.8+) and PyTorch. CEBRA is most effective when used with a GPU, but CPU-only support is provided. We provide instructions to run CEBRA on your system directly. The instructions below were tested on different compute setups with Ubuntu 18.04 or 20.04, using Nvidia GTX 2080, A4000, and V100 cards. Other setups are possible (including Windows), as long as CUDA 10.2+ support is guaranteed.
88

99
- Software dependencies and operating systems:
1010
- Linux or MacOS
@@ -20,25 +20,15 @@ Installation Guide
2020
We outline installation instructions for different systems.
2121
CEBRA will be installed via ``pip install cebra``.
2222

23-
Its dependencies can be installed using ``pip`` or ``conda`` and
23+
CEBRA's dependencies can be installed using ``pip`` or ``conda`` and
2424
we outline different options below.
2525

26-
Most users can only install the **minimal install**. 🚀 For more advanced users, CEBRA has different extra install options that you can select based on your usecase:
27-
28-
* ``[integrations]``: This will install (experimental) support for our streamlit and jupyter integrations.
29-
* ``[docs]``: This will install additional dependencies for building the package documentation.
30-
* ``[dev]``: This will install additional dependencies for development, unit and integration testing,
31-
code formatting, etc. Install this extension if you want to work on a pull request.
32-
* ``[demos]``: This will install additional dependencies for running our demo notebooks.
33-
* ``[datasets]``: This extension will install additional dependencies to use the pre-installed datasets
34-
in ``cebra.datasets``. Note that installing this extension *will not* download the data, which is available on `FigShare <https://figshare.com/s/60adb075234c2cc51fa3>`_.
35-
3626
.. tabs::
3727

3828
.. tab:: Google Colab
3929

40-
CEBRA can also be installed and run on Google colaboratory.
41-
Please see the ``open in colab`` button at the top of each demo notebook for examples.
30+
CEBRA can also be installed and run on Google Colaboratory.
31+
Please see the ``open in colab`` button at the top of each `demo notebook <https://cebra.ai/docs/demos.html>`_ for examples.
4232

4333
If you are starting with a new notebook, simply run
4434

@@ -58,33 +48,6 @@ Most users can only install the **minimal install**. 🚀 For more advanced user
5848
5949
$ conda env create -f conda/cebra.yml
6050
61-
62-
.. tab:: Supplied conda (paper reproduction)
63-
64-
We provide a ``conda`` environment with the full requirements needed to reproduce the first CEBRA paper (although we
65-
recommend using Docker). Namely, you can run CEBRA, piVAE, tSNE and UMAP within this conda env. It is *NOT* needed if you only want to use CEBRA.
66-
67-
* For all platforms except MacOS with M1/2 chipsets, create the full environment using ``cebra_paper.yml``, by running the following from the CEBRA repo root directory:
68-
69-
.. code:: bash
70-
71-
$ conda env create -f conda/cebra_paper.yml
72-
73-
* If you are a MacOS M1 or M2 user and want to reproduce the paper, use the ``cebra_paper_m1.yml`` instead. You'll need to install tensorflow. For that, use `miniconda3 <https://docs.conda.io/projects/conda/en/latest/user-guide/install/macos.html>`_ and follow the setup instructions for tensorflow listed in the `Apple developer docs <https://developer.apple.com/metal/tensorflow-plugin/>`_. In the Terminal, run the following commands:
74-
75-
.. code:: bash
76-
77-
wget https://repo.anaconda.com/miniconda/Miniconda3-py39_4.12.0-MacOSX-arm64.sh -O ~/miniconda.sh
78-
bash ~/miniconda.sh -b -p $HOME/miniconda
79-
source ~/miniconda/bin/activate
80-
conda init zsh
81-
82-
Then, you can build the full environment from the root directory:
83-
84-
.. code:: bash
85-
86-
$ conda env create -f conda/cebra_paper_m1.yml
87-
8851
.. tab:: conda
8952

9053
Conda users should currently use ``pip`` for installation. The missing dependencies will be installed in the install process. A fresh conda environment can be created using
@@ -127,6 +90,16 @@ Most users can only install the **minimal install**. 🚀 For more advanced user
12790
12891
$ pip install cebra
12992
93+
* 🚀 For more advanced users, CEBRA has different extra install options that you can select based on your usecase:
94+
95+
* ``[integrations]``: This will install (experimental) support for our streamlit and jupyter integrations.
96+
* ``[docs]``: This will install additional dependencies for building the package documentation.
97+
* ``[dev]``: This will install additional dependencies for development, unit and integration testing,
98+
code formatting, etc. Install this extension if you want to work on a pull request.
99+
* ``[demos]``: This will install additional dependencies for running our demo notebooks.
100+
* ``[datasets]``: This extension will install additional dependencies to use the pre-installed datasets
101+
in ``cebra.datasets``.
102+
130103
* Inference and development tools only
131104

132105
.. code:: bash
@@ -197,6 +170,39 @@ Installation Troubleshooting
197170

198171
If yopu have issues installing CEBRA, we recommend carefully checking the `traceback`_ which can help you look on `stackoverflow`_ or the popular-in-life-sciences, `Image Forum`_ for similar issues. If you cannot find a solution, please do post an issue on GitHub!
199172

173+
Advanced Installation for Schneider, Lee, Mathis 2023 paper experiments
174+
-----------------------------------------------------------------------
175+
176+
If you want to install the additional dependencies required to run comparisons with other algorithms, please see the following:
177+
178+
.. tabs::
179+
.. tab:: Supplied conda (paper reproduction)
180+
181+
We provide a ``conda`` environment with the full requirements needed to reproduce the first CEBRA paper (although we
182+
recommend using Docker). Namely, you can run CEBRA, piVAE, tSNE and UMAP within this conda env. It is *NOT* needed if you only want to use CEBRA.
183+
184+
* For all platforms except MacOS with M1/2 chipsets, create the full environment using ``cebra_paper.yml``, by running the following from the CEBRA repo root directory:
185+
186+
.. code:: bash
187+
188+
$ conda env create -f conda/cebra_paper.yml
189+
190+
* If you are a MacOS M1 or M2 user and want to reproduce the paper, use the ``cebra_paper_m1.yml`` instead. You'll need to install tensorflow. For that, use `miniconda3 <https://docs.conda.io/projects/conda/en/latest/user-guide/install/macos.html>`_ and follow the setup instructions for tensorflow listed in the `Apple developer docs <https://developer.apple.com/metal/tensorflow-plugin/>`_. In the Terminal, run the following commands:
191+
192+
.. code:: bash
193+
194+
wget https://repo.anaconda.com/miniconda/Miniconda3-py39_4.12.0-MacOSX-arm64.sh -O ~/miniconda.sh
195+
bash ~/miniconda.sh -b -p $HOME/miniconda
196+
source ~/miniconda/bin/activate
197+
conda init zsh
198+
199+
Then, you can build the full environment from the root directory:
200+
201+
.. code:: bash
202+
203+
$ conda env create -f conda/cebra_paper_m1.yml
204+
205+
200206
.. _PyTorch Docs: https://pytorch.org/
201207
.. _virtual environment: https://packaging.python.org/en/latest/guides/installing-using-pip-and-virtual-environments/#creating-a-virtual-environment
202208
.. _traceback: https://realpython.com/python-traceback/

0 commit comments

Comments
 (0)