env.sh

# Based on same file from https://github.com/JiahuiLei/NAP commit 1b062cb
# Original code Copyright (c) 2023 Jiahui Lei. Licensed under the MIT License, see LICENSE_NAP.md
# Modifications Copyright (c) 2025 University of Augsburg (Author: Jens Kreber), licensed under the MIT License, see LICENSE_MIT.md
# Also see NOTICE.md

# Adapted from the original env.sh from NAP

which python
which pip
CC=$CONDA_PREFIX/bin/x86_64-conda-linux-gnu-gcc
CPP=$CONDA_PREFIX/bin/x86_64-conda-linux-gnu-g++
CXX=$CONDA_PREFIX/bin/x86_64-conda-linux-gnu-g++
$CC --version
$CXX --version

# install pytorch
echo ====INSTALLING PyTorch======
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 mkl=2024.0.0  cudatoolkit-dev=11.7* cuda-version=11.7* -c pytorch -c nvidia -y

# install pytorch3d
echo ====INSTALLING=PYTORCH3D======
conda install -c fvcore -c iopath -c conda-forge fvcore iopath -y
conda install pytorch3d=0.7.4 -c pytorch3d -y # 0.6.1

# Install Pytorch Geometry
conda install pyg=2.5.2=py39_torch_2.0.0_cu117 -c pyg -y

# install requirements
pip install cython
pip install -r requirements.txt
pip install numpy==1.23

# build ONet Tools
python setup.py build_ext --inplace