also include nvidia channel for cudatoolkit

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
pytorch · Jun 17, 2021 · 8a6e5e4 · 8a6e5e4
1 parent 5ba192e
commit 8a6e5e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packaging/build_conda.sh b/packaging/build_conda.sh
@@ -11,4 +11,5 @@ setup_conda_pytorch_constraint
 setup_conda_cudatoolkit_constraint
 setup_visual_studio_constraint
 setup_junit_results_folder
-conda build $CONDA_CHANNEL_FLAGS -c defaults --no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchvision
+# nvidia channel included for cudatoolkit >= 11
+conda build $CONDA_CHANNEL_FLAGS -c defaults -c nvidia--no-anaconda-upload --python "$PYTHON_VERSION" packaging/torchvision