-
Notifications
You must be signed in to change notification settings - Fork 654
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
Add CUDA-11.5 builds to torchaudio #2067
Conversation
packaging/build_conda.sh
Outdated
@@ -11,4 +11,4 @@ setup_conda_pytorch_constraint | |||
setup_conda_cudatoolkit_constraint | |||
setup_visual_studio_constraint | |||
# nvidia channel included for cudatoolkit >= 11 | |||
conda build -c defaults -c nvidia $CONDA_CHANNEL_FLAGS --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchaudio | |||
conda build -c defaults -c conda-forge $CONDA_CHANNEL_FLAGS --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchaudio |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's do the same change with limiting conda-forge
to only cuda 11.5 as we did in vision
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes totally!
@atalman has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Summary: cc peterjc123 maxluk nbcsm guyang3532 gunandrose4u smartcat2010 mszhanyi Pull Request resolved: pytorch#2067 Reviewed By: seemethere Differential Revision: D33032607 Pulled By: atalman fbshipit-source-id: a5767e9af27690d3a7ab762ddf30178b3069cd35
By adding pygame to the requirements and bumping gym to 0.25.0 (to account for changes introduced by pytorch/tutorials#2026 ) Add cell for installing `gym` on Colab
cc @peterjc123 @maxluk @nbcsm @guyang3532 @gunandrose4u @smartcat2010 @mszhanyi