- Python 3.9
-
Install poetry
If you don't have
poetry
, get by running following command.
โป Do not usepip
to installpoetry
.curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/get-poetry.py | python -
-
Activate poetry environment
poetry shell
-
Install dependencies
poetry install
-
Install Pytorch
Install PyTorch that works in your environmen. Normally, adding by following command.
poetry add torch torchvision
Or, if your GPU is NVIDIA RTX A6000 or something like "sm_86" architecture and CUDA version is 11.7, you have to adding by following command.
poe pytorch