Install PyTorch without CUDA
Since Leapcell machines do not have GPUs, we need to install the version of PyTorch without CUDA support. This helps reduce the image size.
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu