Install Conda
conda create --name stegastamp python=3.10.0
source activate stegastamp
pip install -r requirements.txt
Install PyTorch
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
Install Pytoch-lightning
pip install pytorch-lightning