Installation

Prerequisites

Create a conda virtual environment and activate it

$ conda create -n calcil python=3.9
$ conda activate calcil

Install CUDA and cuDNN in conda virtual env (you may opt to skip this step if you have CUDA installed in your system and you know what you are doing)
```
$ conda install -c conda-forge cudatoolkit~=11.8.0 cudnn~=8.8.0
$ conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
```
Install jaxlib. Note that the following command is for CUDA 11.x and cuDNN 8.2+. If you have different versions of CUDA, please refer to JAX installation guide and make sure to match the version numbers of jaxlib and jax (as specified in requirements.txt).
```
$ pip install jaxlib==0.3.18+cuda11.cudnn82 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
```
Note

To test the installation of jax, you can run the following command:
```
$ python -c "import jax.numpy as jnp; print(jnp.ones(5)+jnp.zeros(5))"
```
pip install CalCIL. You may use -e flag to install in editable mode.
```
$ pip install git+https://github.com/rmcao/CalCIL.git
```
Install optional dependencies for interactive visualization via Jupyter lab
```
$ conda install -c conda-forge jupyterlab nodejs ipympl
```