Add -c nweidia which contains pytorch-cu124

nWEIdia · nWEIdia · commit ccdfe5c5d539 · 2024-04-22T13:23:57.000-07:00
diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
@@ -413,7 +413,7 @@ for py_ver in "${DESIRED_PYTHON[@]}"; do
         else
           local_channel="$(pwd)/$output_folder"
         fi
-        conda install -y -c "file://$local_channel" pytorch==$PYTORCH_BUILD_VERSION -c pytorch -c numba/label/dev -c pytorch-nightly -c nvidia
+        conda install -y -c "file://$local_channel" pytorch==$PYTORCH_BUILD_VERSION -c pytorch -c numba/label/dev -c pytorch-nightly -c nvidia -c nweidia
 
         echo "$(date) :: Running tests"
         pushd "$pytorch_rootdir"
