주요 콘텐츠로 건너뛰기

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