From 5f43cf63571b5d9b64da33f82642be338230258e Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 13 Mar 2023 13:30:30 -0700 Subject: [PATCH] Add pytorch conda dependencies --- release/promote.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/release/promote.sh b/release/promote.sh index 8605682f9..f6ef71ebd 100644 --- a/release/promote.sh +++ b/release/promote.sh @@ -104,6 +104,10 @@ promote_s3 "requests" whl "2.28.1" # promote_s3 torchdata whl "${TORCHDATA_VERSION}" # promote_s3 "libtorch-*" libtorch "${PYTORCH_VERSION}" +promote_conda torchtriton conda "2.0.0" +promote_conda pytorch-cuda conda "11.7" +promote_conda pytorch-cuda conda "11.8" + # promote_conda pytorch conda "${PYTORCH_VERSION}" # promote_conda torchvision conda "${TORCHVISION_VERSION}" # promote_conda torchaudio conda "${TORCHAUDIO_VERSION}"