Skip to main content

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