TorchScript bad_alloc issue (pytorch#2542)

Summary: Pull Request resolved: pytorch#2542 Differential Revision: D65495806
PaulZhang12 · Nov 5, 2024 · 51fdf03 · 51fdf03
1 parent 7e867ad
commit 51fdf03
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/.github/workflows/unittest_ci_cpu.yml b/.github/workflows/unittest_ci_cpu.yml
@@ -45,9 +45,8 @@ jobs:
         conda info
         python --version
         conda run -n build_binary python --version
-        conda install -n build_binary \
-          --yes \
-          pytorch cpuonly -c pytorch-nightly
+        conda run -n build_binary \
+          pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
         conda run -n build_binary \
           python -c "import torch.distributed"
         conda run -n build_binary \