1 2
how to install pytorch on conda with gpu mamba install pytorch=1.12 torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia -c conda-forge