Skip to content

Simplify install: add PyTorch cu130 repo for torch/vision/audio#24

Open
wachawo wants to merge 4 commits intoNVIDIA:mainfrom
wachawo:cu130
Open

Simplify install: add PyTorch cu130 repo for torch/vision/audio#24
wachawo wants to merge 4 commits intoNVIDIA:mainfrom
wachawo:cu130

Conversation

@wachawo
Copy link
Contributor

@wachawo wachawo commented Jan 24, 2026

Changes:

  • Added PyTorch wheel repository https://download.pytorch.org/whl/cu130 to make installing torch, torchvision, and torchaudio easier and to simplify Docker image builds.

  • The installation for macOS is left with pypi.org.

  • Added package accelerate, which is required when running with the --cpu-offload flag.

ImportError: CPU offloading requires the 'accelerate' package. Install it with: pip install accelerate

Possible installation options:

docker compose build
pip install -r moshi/requirements.txt
uv pip install moshi/.
pip install moshi/.
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130

These changes simplify and automate the deployment process.

Fix #23 #31

@CharlesMcquade
Copy link

Definitely need this one ASAP

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Docker: RTX 5090 (Blackwell, sm_120) not supported by bundled PyTorch → “no kernel image is available”

2 participants