Welcome to the repository of Pty-chi, a PyTorch-based ptychography reconstruction library!
The easiest way to install the latest release is through PyPI.
First, create a new conda environment with Python 3.11:
conda create -n ptychi python=3.11
Then install Pty-Chi using:
pip install ptychi
Use developer installation when you want to modify the code and test the changes, or when you run into build issues that drive you to install the package from source. We recommend using Conda/pip or uv for environment and package management.
To install the latest code in the main branch, clone the repository to your workspace, and create a new conda environment using:
conda create -n ptychi -c conda-forge -c nvidia --file requirements.txt
Then install the package using:
pip install -e .
Uv is a modern lightweight package manager for Python featuring fast speed and
deterministic builds. When creating a uv virtual environment, the environment
directory and all the packages inatalled in it are kept in the current working
directory -- unlike Conda, where the environments are centrally managed. Therefore,
first cd
into the root level of your local clone of the repository, and then create
a new uv virtual environment with Python 3.11:
uv venv --python 3.11 .venv
Activate the environment:
source .venv/bin/activate
Then install Pty-Chi and its dependencies using:
uv pip install -r requirements.txt uv pip install -e .
- Contact the developers to be given access to the APS GitLab repository that holds test data. You need to have an account on APS GitLab.
- After gaining access, clone the GitLab data repository to your hard drive.
- Set
PTYCHO_CI_DATA_DIR
to theci_data
directory of the data repository:export PTYCHO_CI_DATA_DIR="path_to_data_repo/ci_data"
. - Run any test scripts in
tests
with Python.
Pty-Chi's documentation is hosted on Read the Docs.
You can also build the docs and view them in your browser locally. To build the docs, install the dependencies as the first step:
pip install -r docs/requirements.txt
Then:
cd docs make html
You can then view the docs by opening docs/build/html/index.html
in your browser.
Please refer to the developer's guide for more information on how to contribute to the project. The developer's guide is hosted on the Wiki page of Pty-Chi's APS GitLab repository. To gain access to the APS GitLab repository, please contact the developers.