git clone --recurse-submodules [email protected]:yanyongyu/pytorch-learning.git
To create an python environment using conda
and poetry
:
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