Project Repository for the ML4Earth Project
GPU enabled, no rapids
pip install -r requirements.txt
- NVIDIA RAPIDS Setup based on PyTorch
- https://catalog.ngc.nvidia.com/orgs/nvidia/teams/rapidsai/containers/rapidsai
- Base Image:
nvcr.io/nvidia/pytorch:23.03-py3
- Everything should be compose oriented