Create your virtual environment of choice conda create --name sfti-env python=3.8 conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia