wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run && \
sudo sh cuda_12.4.0_550.54.14_linux.run && \
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124