Skip to content

Remove torchaudio & torchvision deps - #53

Open
sbonner0 wants to merge 1 commit into
NVIDIA-BioNeMo:devfrom
sbonner0:dev
Open

sbonner0 wants to merge 1 commit into
NVIDIA-BioNeMo:devfrom
sbonner0:dev

Conversation

@sbonner0

Copy link
Copy Markdown

Summary

Removes torchvision and torchaudio from the PyTorch install step in env/build_uv_env.sh. Neither package is used anywhere in the repo, and their presence was causing an import-time crash.

Fixes #42

Problem

Fresh environments crashed on import with a torchaudio/torch version mismatch. The install line pinned torch==2.7.0+cu126 but left torchvision and torchaudio unpinned, so the resolver pulled the latest builds — compiled against a different torch version, producing an ABI mismatch.

Fix

-uv pip install torch==2.7.0+cu126 torchvision torchaudio \
+uv pip install torch==2.7.0+cu126 \
     --index-url https://download.pytorch.org/whl/cu126
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant