Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update torchaudio in Dockerfile to match torch version #3637

Merged
merged 2 commits into from
Feb 18, 2022

Conversation

GNroy
Copy link
Collaborator

@GNroy GNroy commented Feb 10, 2022

Signed-off-by: Aleksandr Laptev [email protected]

What does this PR do ?

This PR fixes the version mismatch between torchaudio and torch (0.10 vs 1.11) in Dockerfile.
If the version of torchaudio does not match the version of torch, then it loses its CUDA backend and the following errors may occur:

NotImplementedError: Could not run 'torchaudio::rnnt_loss' with arguments from the 'CUDA' backend.

Changelog

Dockerfile

  • --branch release/0.10 -> --branch release/0.11

@titu1994 titu1994 merged commit c783f7f into NVIDIA:main Feb 18, 2022
fayejf pushed a commit that referenced this pull request Mar 2, 2022
Signed-off-by: Aleksandr Laptev <[email protected]>

Co-authored-by: Aleksandr Laptev <[email protected]>
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.

2 participants