diff --git a/.github/workflows/aistore_ci.yml b/.github/workflows/aistore_ci.yml index f5b0abb2d..acf15c20a 100644 --- 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 index d31b17556..643334ebd 100644 --- 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 index af1bfaff4..06bcef7cd 100644 --- 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 index 07070038c..ce720380c 100644 --- 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 index e77158c98..77433bf51 100644 --- 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 index b083b40a9..5ef46a5b0 100644 --- 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 index 6437cc5b4..dd3631b4d 100644 --- 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/+.\+//')"