Template for my research projects using PyTorch
.github/test.yaml # pytest and yamllint
.gitignore
notebooks # Jupyter notebooks
script # Shell scripts for running experiments
src # Root of main codes
tests
README.md # This file
I usually use conda to create virtual environments for PyTorch.
conda create -n ${PROJECT_NAME} python=3.10
conda activate ${PROJECT_NAME}
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip install -U hatch
hatch run tests:pytest
pip install -U hatch
hatch run docs:build
...
...
...
...