Skip to content

Commit

Permalink
Andrewkh/update nightly channel (#1336)
Browse files Browse the repository at this point in the history
* update nightly channel spec and readmes
  • Loading branch information
andrewkho authored Oct 16, 2024
1 parent 6b762ba commit b7d5dc5
Show file tree
Hide file tree
Showing 7 changed files with 15 additions and 15 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/aistore_ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/lint.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/stateful_dataloader_ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion CONTRIBUTING.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,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:
Expand Down
2 changes: 1 addition & 1 deletion packaging/pkg_helpers.bash
Original file line number Diff line number Diff line change
Expand Up @@ -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/+.\+//')"
Expand Down

0 comments on commit b7d5dc5

Please sign in to comment.