Andrewkh/update nightly channel (#1336)

* update nightly channel spec and readmes
pytorch · Oct 16, 2024 · f3e8fc3 · f3e8fc3
1 parent c4f57b4
commit f3e8fc3
Show file tree

Hide file tree

Showing 7 changed files with 15 additions and 15 deletions.
diff --git a/.github/workflows/aistore_ci.yml b/.github/workflows/aistore_ci.yml
@@ -30,9 +30,9 @@ jobs:
         shell: bash
         run: |
           if [[ "${{ github.base_ref }}" == release/* ]] || [[ "${{ github.ref }}" == refs/heads/release/* ]] || [[ "${{ github.ref }}" == refs/tags/v* ]]; then
-            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu/torch_test.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu"
           else
-            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu"
           fi
           echo "value=$PT_CHANNEL" >> $GITHUB_OUTPUT
         id: pytorch_channel
@@ -45,7 +45,7 @@ jobs:
       - name: Install dependencies
         run: |
           pip3 install -r requirements.txt
-          pip3 install --pre torch -f "${{ steps.pytorch_channel.outputs.value }}"
+          pip3 install --pre torch --index-url "${{ steps.pytorch_channel.outputs.value }}"
       - name: Run AIStore local deployment
         uses: NVIDIA/aistore@main
       - name: Build TorchData

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -34,9 +34,9 @@ jobs:
         shell: bash
         run: |
           if [[ "${{ github.base_ref }}" == release/* ]] || [[ "${{ github.ref }}" == refs/heads/release/* ]] || [[ "${{ github.ref }}" == refs/tags/v* ]]; then
-            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu/torch_test.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu"
           else
-            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu"
           fi
           echo "value=$PT_CHANNEL" >> $GITHUB_OUTPUT
         id: pytorch_channel
@@ -66,7 +66,7 @@ jobs:
         run: |
           pip3 install -r requirements.txt
           pip3 install networkx
-          pip3 install --pre torch -f "${{ steps.pytorch_channel.outputs.value }}"
+          pip3 install --pre torch --index-url "${{ steps.pytorch_channel.outputs.value }}"
           pip3 install cmake ninja
           echo "/home/runner/.local/bin" >> $GITHUB_PATH
       - name: Build TorchData

diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
@@ -36,9 +36,9 @@ jobs:
         shell: bash
         run: |
           if [[ "${{ github.base_ref }}" == release/* ]] || [[ "${{ github.ref }}" == refs/heads/release/* ]] || [[ "${{ github.ref }}" == refs/tags/v* ]]; then
-            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu/torch_test.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu"
           else
-            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu"
           fi
           echo "value=$PT_CHANNEL" >> $GITHUB_OUTPUT
         id: pytorch_channel
@@ -51,7 +51,7 @@ jobs:
       - name: Install PyTorch
         run: |
           pip3 install networkx
-          pip3 install --pre torch -f "${{ steps.pytorch_channel.outputs.value }}"
+          pip3 install --pre torch --index-url "${{ steps.pytorch_channel.outputs.value }}"
       - name: Install dependencies
         run: |
           pip3 install -r requirements.txt

diff --git a/.github/workflows/stateful_dataloader_ci.yml b/.github/workflows/stateful_dataloader_ci.yml
@@ -34,9 +34,9 @@ jobs:
         shell: bash
         run: |
           if [[ "${{ github.base_ref }}" == release/* ]] || [[ "${{ github.ref }}" == refs/heads/release/* ]] || [[ "${{ github.ref }}" == refs/tags/v* ]]; then
-            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu/torch_test.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/test/cpu"
           else
-            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html"
+            PT_CHANNEL="https://download.pytorch.org/whl/nightly/cpu"
           fi
           echo "value=$PT_CHANNEL" >> $GITHUB_OUTPUT
         id: pytorch_channel
@@ -66,7 +66,7 @@ jobs:
         run: |
           pip3 install -r requirements.txt
           pip3 install networkx
-          pip3 install --pre torch -f "${{ steps.pytorch_channel.outputs.value }}"
+          pip3 install --pre torch --index-url "${{ steps.pytorch_channel.outputs.value }}"
           pip3 install cmake ninja
           echo "/home/runner/.local/bin" >> $GITHUB_PATH
       - name: Build TorchData

diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
@@ -43,7 +43,7 @@ For question related to the usage of this library, please post a question on the
 conda install pytorch -c pytorch-nightly
 # or with pip (see https://pytorch.org/get-started/locally/)
 # pip install numpy
-# pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+# pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
 ```
 
 ### Install TorchData

diff --git a/README.md b/README.md
@@ -98,7 +98,7 @@ The nightly version of TorchData is also provided and updated daily from main br
 Using pip:
 
 ```bash
-pip install --pre torchdata --extra-index-url https://download.pytorch.org/whl/nightly/cpu
+pip install --pre torchdata --index-url https://download.pytorch.org/whl/nightly/cpu
 ```
 
 Using conda:

diff --git a/packaging/pkg_helpers.bash b/packaging/pkg_helpers.bash
@@ -135,7 +135,7 @@ setup_pip_pytorch_version() {
   if [[ -z "$PYTORCH_VERSION" ]]; then
     # Install latest prerelease version of torch, per our nightlies, consistent
     # with the requested cuda version
-    pip_install --pre torch -f "https://download.pytorch.org/whl/nightly/${WHEEL_DIR}"
+    pip_install --pre torch --index-url "https://download.pytorch.org/whl/nightly/${WHEEL_DIR}"
     # CUDA and CPU are ABI compatible on the CPU-only parts, so strip
     # in this case
     export PYTORCH_VERSION="$(pip show torch | grep ^Version: | sed 's/Version:  *//' | sed 's/+.\+//')"