Skip to content

Installation Guideยค

clone the repo
git clone https://github.com/microsoft/ClimaX
create and activate env
cd ClimaX
conda env create --file docker/environment.yml
conda activate climaX
install this package
# install so the project is in PYTHONPATH
pip install -e .
build docker container
cd docker
docker build -t ClimaX .
run docker container
cd ClimaX
docker run --gpus all -it --rm --user $(id -u):$(id -g) \
    -v $(pwd):/code -v /mnt/data:/data --workdir /code -e PYTHONPATH=/code/src \
    ClimaX:latest

Note

  • --gpus all -it --rm --user $(id -u):$(id -g): enables using all GPUs and runs an interactive session with current user's UID/GUID to avoid docker writing files as root.
  • -v $(pwd):/code -v /mnt/data:/data --workdir /code: mounts current directory and data directory (i.e. the cloned git repo) to /code and /data respectively, and use the code directory as the current working directory.