git clone --recurse-submodules [email protected]:yanyongyu/pytorch-learning.git

conda create -n pytorch python=3.8  # or higher
conda activate pytorch

# install pytorch by conda
# https://pytorch.org/get-started/locally/#start-locally

# CUDA 10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
# CUDA 11.1
# conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

# install base dependencies
poetry install
# install extra dependencies: jupyter notebook
# poetry install -E jupyter