https://pytorch.org/get-started/locally/
1.1 Get cuda version
1.2 Install Cuda or upgrade
Need cuda 12.4
So i want to upgrade cuda
Go to https://developer.nvidia.com/cuda-downloads
https://developer.nvidia.com/cuda-12-4-0-download-archive
1.3 Install pytorch
bash
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124