С официального сайта ставим CUDA (предварительно убедившись, что видеокарта поддерживает это)
https://developer.nvidia.com/cuda-toolkit-archive
Примечание: на данный момент pytorch поддерживает версию 11.8
После установки Nvidia Cuda переходим на сайт https://pytorch.org/get-started/previous-versions/
conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidiaДалее
pip install cuda-python
