... | ... | @@ -69,7 +69,35 @@ All thanks to Andreas Herten (@herten1) for installing these modules and the man |
|
|
|
|
|
### Pytorch & HEAT
|
|
|
|
|
|
tbd: Bjoern Hagemeier @hagemeier2
|
|
|
The Helmholtz Analytics Toolkit ([HeAT](https://github.com/helmholtz-analytics/heat)) is based on [PyTorch](https://pytorch.org/). Documentation, including tutorials, is hosted at [ReadTheDocs](https://heat.readthedocs.io/en/latest/).
|
|
|
|
|
|
```shell
|
|
|
module use $OTHERSTAGES
|
|
|
ml Stages/2020
|
|
|
ml GCC/9.3.0 CUDA/11.0 Python/3.8.5
|
|
|
ml ParaStationMPI/5.4.7-1 UCX/1.9.0 mpi4py/3.0.3-Python-3.8.5
|
|
|
ml HDF5/1.10.6 netCDF/4.7.4
|
|
|
ml NCCL/2.8.3-1-CUDA-11.0
|
|
|
ml SciPy-Stack/2020-Python-3.8.5
|
|
|
ml parallel-netcdf/1.12.1
|
|
|
|
|
|
unset UCX_TLS
|
|
|
export PSP_CUDA=0
|
|
|
```
|
|
|
You can then setup a virtual environment and install HeAT in turn:
|
|
|
```shell
|
|
|
python -m venv heat-venv
|
|
|
source heat-venv/bin/activate
|
|
|
pip install 'heat[hdf5,netcdf]'
|
|
|
```
|
|
|
Or, if you want the latest, greatest version, install from the sources:
|
|
|
```shell
|
|
|
python -m venv heat-venv
|
|
|
source heat-venv/bin/activate
|
|
|
git clone https://github.com/helmholtz-analytics/heat
|
|
|
cd heat
|
|
|
pip install '.[hdf5,netcdf]'
|
|
|
```
|
|
|
|
|
|
# Example Projects
|
|
|
|
... | ... | |