[LTBS] Pytorch CUDA NVIDIA Driver

ubuntu <= 22.04 , cuda 12.4 , driver 550

 

wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run

 

sudo sh cuda_12.4.1_550.54.15_linux.run

 

pip3 install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124