Skip to content

Commit a0134d2

Browse files
Bordalantigapre-commit-ci[bot]
authored
bump: testing with latest torch 2.6 (#20509)
* bump: testing with future torch 2.6 * bump `typing-extensions` * TORCHINDUCTOR_CACHE_DIR * bitsandbytes * Apply suggestions from code review * _TORCH_LESS_EQUAL_2_6 --------- Co-authored-by: Luca Antiga <luca.antiga@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Luca Antiga <luca@lightning.ai>
1 parent 02e2668 commit a0134d2

File tree

24 files changed

+102
-250
lines changed

24 files changed

+102
-250
lines changed

‎.azure/gpu-tests-fabric.yml

+12-7
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,14 @@ jobs:
5656
options: "--gpus=all --shm-size=2gb -v /var/tmp:/var/tmp"
5757
strategy:
5858
matrix:
59+
"Fabric | oldest":
60+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.1"
61+
PACKAGE_NAME: "fabric"
5962
"Fabric | latest":
60-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.3-cuda12.1.0"
63+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
6164
PACKAGE_NAME: "fabric"
6265
"Lightning | latest":
63-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.5-cuda12.1.0"
66+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
6467
PACKAGE_NAME: "lightning"
6568
workspace:
6669
clean: all
@@ -77,9 +80,8 @@ jobs:
7780
displayName: "set env. vars"
7881
- bash: |
7982
echo "##vso[task.setvariable variable=TORCH_URL]https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM}"
80-
echo "##vso[task.setvariable variable=TORCHVISION_URL]https://download.pytorch.org/whl/test/cu124/torchvision-0.19.0%2Bcu124-cp${PYTHON_VERSION_MM}-cp${PYTHON_VERSION_MM}-linux_x86_64.whl"
8183
condition: endsWith(variables['Agent.JobName'], 'future')
82-
displayName: "set env. vars 4 future"
84+
displayName: "extend env. vars 4 future"
8385
8486
- bash: |
8587
echo $(DEVICES)
@@ -105,15 +107,17 @@ jobs:
105107
displayName: "Adjust dependencies"
106108
107109
- bash: |
110+
set -e
108111
extra=$(python -c "print({'lightning': 'fabric-'}.get('$(PACKAGE_NAME)', ''))")
109-
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
112+
pip install -e ".[${extra}dev]" pytest-timeout -U --extra-index-url="${TORCH_URL}"
110113
pip install setuptools==75.6.0 jsonargparse==4.35.0
111114
displayName: "Install package & dependencies"
112115
113116
- bash: |
114117
set -e
115118
python requirements/collect_env_details.py
116119
python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu == 2, f'GPU: {mgpu}'"
120+
python requirements/pytorch/check-avail-extras.py
117121
python -c "import bitsandbytes"
118122
displayName: "Env details"
119123
@@ -140,10 +144,11 @@ jobs:
140144
displayName: "Testing: fabric standard"
141145
timeoutInMinutes: "10"
142146

143-
- bash: bash ./run_standalone_tests.sh "tests_fabric"
147+
- bash: |
148+
wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/run_standalone_tests.sh
149+
bash ./run_standalone_tests.sh "tests_fabric"
144150
workingDirectory: tests/
145151
env:
146-
PL_STANDALONE_TESTS_SOURCE: $(COVERAGE_SOURCE)
147152
PL_RUN_STANDALONE_TESTS: "1"
148153
displayName: "Testing: fabric standalone"
149154
timeoutInMinutes: "10"

‎.azure/gpu-tests-pytorch.yml

+11-7
Original file line numberDiff line numberDiff line change
@@ -49,11 +49,14 @@ jobs:
4949
cancelTimeoutInMinutes: "2"
5050
strategy:
5151
matrix:
52+
"PyTorch | oldest":
53+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.1-cuda12.1.1"
54+
PACKAGE_NAME: "pytorch"
5255
"PyTorch | latest":
53-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.11-torch2.3-cuda12.1.0"
56+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
5457
PACKAGE_NAME: "pytorch"
5558
"Lightning | latest":
56-
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.5-cuda12.1.0"
59+
image: "pytorchlightning/pytorch_lightning:base-cuda-py3.12-torch2.6-cuda12.4.1"
5760
PACKAGE_NAME: "lightning"
5861
pool: lit-rtx-3090
5962
variables:
@@ -81,9 +84,8 @@ jobs:
8184
displayName: "set env. vars"
8285
- bash: |
8386
echo "##vso[task.setvariable variable=TORCH_URL]https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM}"
84-
echo "##vso[task.setvariable variable=TORCHVISION_URL]https://download.pytorch.org/whl/test/cu124/torchvision-0.19.0%2Bcu124-cp${PYTHON_VERSION_MM}-cp${PYTHON_VERSION_MM}-linux_x86_64.whl"
8587
condition: endsWith(variables['Agent.JobName'], 'future')
86-
displayName: "set env. vars 4 future"
88+
displayName: "extend env. vars 4 future"
8789
8890
- bash: |
8991
echo $(DEVICES)
@@ -109,8 +111,9 @@ jobs:
109111
displayName: "Adjust dependencies"
110112
111113
- bash: |
114+
set -e
112115
extra=$(python -c "print({'lightning': 'pytorch-'}.get('$(PACKAGE_NAME)', ''))")
113-
pip install -e ".[${extra}dev]" pytest-timeout -U --find-links="${TORCH_URL}" --find-links="${TORCHVISION_URL}"
116+
pip install -e ".[${extra}dev]" pytest-timeout -U --extra-index-url="${TORCH_URL}"
114117
pip install setuptools==75.6.0 jsonargparse==4.35.0
115118
displayName: "Install package & dependencies"
116119
@@ -161,11 +164,12 @@ jobs:
161164
displayName: "Testing: PyTorch standard"
162165
timeoutInMinutes: "35"
163166

164-
- bash: bash ./run_standalone_tests.sh "tests_pytorch"
167+
- bash: |
168+
wget https://raw.githubusercontent.com/Lightning-AI/utilities/main/scripts/run_standalone_tests.sh
169+
bash ./run_standalone_tests.sh "tests_pytorch"
165170
workingDirectory: tests/
166171
env:
167172
PL_USE_MOCKED_MNIST: "1"
168-
PL_STANDALONE_TESTS_SOURCE: $(COVERAGE_SOURCE)
169173
PL_RUN_STANDALONE_TESTS: "1"
170174
displayName: "Testing: PyTorch standalone tests"
171175
timeoutInMinutes: "35"

‎.github/actions/pip-wheels/action.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ runs:
4646
run: |
4747
# cat requirements.dump
4848
pip wheel -r requirements.dump --prefer-binary \
49-
--wheel-dir=.wheels \
50-
-f ${{ inputs.torch-url }} -f ${{ inputs.wheel-dir }}
49+
--wheel-dir=".wheels" \
50+
--extra-index-url=${{ inputs.torch-url }} -f ${{ inputs.wheel-dir }}
5151
ls -lh .wheels/
5252
shell: bash
5353

‎.github/workflows/_legacy-checkpoints.yml

+3-3
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ on:
4343

4444
env:
4545
LEGACY_FOLDER: "tests/legacy"
46-
TORCH_URL: "https://download.pytorch.org/whl/cpu/torch_stable.html"
46+
TORCH_URL: "https://download.pytorch.org/whl/cpu/"
4747

4848
defaults:
4949
run:
@@ -67,12 +67,12 @@ jobs:
6767
PACKAGE_NAME: pytorch
6868
FREEZE_REQUIREMENTS: 1
6969
timeout-minutes: 20
70-
run: pip install . -f ${TORCH_URL}
70+
run: pip install . --extra-index-url="${TORCH_URL}"
7171
if: inputs.pl_version == ''
7272

7373
- name: Install PL version
7474
timeout-minutes: 20
75-
run: pip install "pytorch-lightning==${{ inputs.pl_version }}" -f ${TORCH_URL}
75+
run: pip install "pytorch-lightning==${{ inputs.pl_version }}" --extra-index-url="${TORCH_URL}"
7676
if: inputs.pl_version != ''
7777

7878
- name: Adjust tests -> PL

‎.github/workflows/ci-tests-fabric.yml

+19-27
Original file line numberDiff line numberDiff line change
@@ -56,36 +56,28 @@ jobs:
5656
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.5.1" }
5757
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.5.1" }
5858
# only run PyTorch latest with Python latest, use Fabric scope to limit dependency issues
59-
- { os: "macOS-14", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.5.1" }
60-
- { os: "ubuntu-22.04", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.5.1" }
61-
- { os: "windows-2022", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.5.1" }
59+
- { os: "macOS-14", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.6" }
60+
- { os: "ubuntu-22.04", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.6" }
61+
- { os: "windows-2022", pkg-name: "fabric", python-version: "3.12.7", pytorch-version: "2.6" }
6262
# "oldest" versions tests, only on minimum Python
63-
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.9", pytorch-version: "2.1", requires: "oldest" }
64-
- {
65-
os: "ubuntu-20.04",
66-
pkg-name: "lightning",
67-
python-version: "3.9",
68-
pytorch-version: "2.1",
69-
requires: "oldest",
70-
}
71-
- {
72-
os: "windows-2022",
73-
pkg-name: "lightning",
74-
python-version: "3.9",
75-
pytorch-version: "2.1",
76-
requires: "oldest",
77-
}
63+
- { os: "macOS-14", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
64+
- { os: "ubuntu-20.04", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
65+
- { os: "windows-2022", pkg-name: "fabric", pytorch-version: "2.1", requires: "oldest" }
7866
# "fabric" installs the standalone package
79-
- { os: "macOS-14", pkg-name: "fabric", python-version: "3.9", pytorch-version: "2.1" }
80-
- { os: "ubuntu-20.04", pkg-name: "fabric", python-version: "3.9", pytorch-version: "2.1" }
81-
- { os: "windows-2022", pkg-name: "fabric", python-version: "3.9", pytorch-version: "2.1" }
67+
- { os: "macOS-14", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
68+
- { os: "ubuntu-20.04", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
69+
- { os: "windows-2022", pkg-name: "fabric", python-version: "3.10", pytorch-version: "2.5" }
70+
# adding recently cut Torch 2.7 - FUTURE
71+
# - { os: "macOS-14", pkg-name: "fabric", python-version: "3.12", pytorch-version: "2.7" }
72+
# - { os: "ubuntu-22.04", pkg-name: "fabric", python-version: "3.12", pytorch-version: "2.7" }
73+
# - { os: "windows-2022", pkg-name: "fabric", python-version: "3.12", pytorch-version: "2.7" }
8274
timeout-minutes: 25 # because of building grpcio on Mac
8375
env:
8476
PACKAGE_NAME: ${{ matrix.pkg-name }}
8577
FREEZE_REQUIREMENTS: ${{ ! (github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/heads/release/')) }}
8678
PYPI_CACHE_DIR: "_pip-wheels"
87-
TORCH_URL_STABLE: "https://download.pytorch.org/whl/cpu/torch_stable.html"
88-
TORCH_URL_TEST: "https://download.pytorch.org/whl/test/cpu/torch"
79+
TORCH_URL_STABLE: "https://download.pytorch.org/whl/cpu/"
80+
TORCH_URL_TEST: "https://download.pytorch.org/whl/test/cpu/"
8981
# TODO: Remove this - Enable running MPS tests on this platform
9082
DISABLE_MPS: ${{ matrix.os == 'macOS-14' && '1' || '0' }}
9183
steps:
@@ -94,7 +86,7 @@ jobs:
9486
- name: Set up Python ${{ matrix.python-version }}
9587
uses: actions/setup-python@v5
9688
with:
97-
python-version: ${{ matrix.python-version }}
89+
python-version: ${{ matrix.python-version || '3.9' }}
9890

9991
- name: basic setup
10092
run: pip install -q -r .actions/requirements.txt
@@ -126,8 +118,8 @@ jobs:
126118
127119
- name: Env. variables
128120
run: |
129-
# Switch PyTorch URL
130-
python -c "print('TORCH_URL=' + str('${{env.TORCH_URL_TEST}}' if '${{ matrix.pytorch-version }}' == '2.5' else '${{env.TORCH_URL_STABLE}}'))" >> $GITHUB_ENV
121+
# Switch PyTorch URL between stable and test/future
122+
python -c "print('TORCH_URL=' + str('${{env.TORCH_URL_TEST}}' if '${{ matrix.pytorch-version }}' == '2.7' else '${{env.TORCH_URL_STABLE}}'))" >> $GITHUB_ENV
131123
# Switch coverage scope
132124
python -c "print('COVERAGE_SCOPE=' + str('lightning' if '${{matrix.pkg-name}}' == 'lightning' else 'lightning_fabric'))" >> $GITHUB_ENV
133125
# if you install mono-package set dependency only for this subpackage
@@ -139,7 +131,7 @@ jobs:
139131
timeout-minutes: 20
140132
run: |
141133
pip install -e ".[${EXTRA_PREFIX}test,${EXTRA_PREFIX}strategies]" -U --prefer-binary \
142-
--find-links="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}"
134+
--extra-index-url="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}"
143135
pip list
144136
- name: Dump handy wheels
145137
if: github.event_name == 'push' && github.ref == 'refs/heads/master'

‎.github/workflows/ci-tests-pytorch.yml

+20-28
Original file line numberDiff line numberDiff line change
@@ -60,35 +60,27 @@ jobs:
6060
- { os: "ubuntu-22.04", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.5.1" }
6161
- { os: "windows-2022", pkg-name: "lightning", python-version: "3.12.7", pytorch-version: "2.5.1" }
6262
# only run PyTorch latest with Python latest, use PyTorch scope to limit dependency issues
63-
- { os: "macOS-14", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.5.1" }
64-
- { os: "ubuntu-22.04", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.5.1" }
65-
- { os: "windows-2022", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.5.1" }
63+
- { os: "macOS-14", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.6" }
64+
- { os: "ubuntu-22.04", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.6" }
65+
- { os: "windows-2022", pkg-name: "pytorch", python-version: "3.12.7", pytorch-version: "2.6" }
6666
# "oldest" versions tests, only on minimum Python
67-
- { os: "macOS-14", pkg-name: "lightning", python-version: "3.9", pytorch-version: "2.1", requires: "oldest" }
68-
- {
69-
os: "ubuntu-20.04",
70-
pkg-name: "lightning",
71-
python-version: "3.9",
72-
pytorch-version: "2.1",
73-
requires: "oldest",
74-
}
75-
- {
76-
os: "windows-2022",
77-
pkg-name: "lightning",
78-
python-version: "3.9",
79-
pytorch-version: "2.1",
80-
requires: "oldest",
81-
}
67+
- { os: "macOS-14", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
68+
- { os: "ubuntu-20.04", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
69+
- { os: "windows-2022", pkg-name: "pytorch", pytorch-version: "2.1", requires: "oldest" }
8270
# "pytorch" installs the standalone package
83-
- { os: "macOS-14", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "2.1" }
84-
- { os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "2.1" }
85-
- { os: "windows-2022", pkg-name: "pytorch", python-version: "3.9", pytorch-version: "2.1" }
71+
- { os: "macOS-14", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
72+
- { os: "ubuntu-20.04", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
73+
- { os: "windows-2022", pkg-name: "pytorch", python-version: "3.10", pytorch-version: "2.5" }
74+
# adding recently cut Torch 2.7 - FUTURE
75+
# - { os: "macOS-14", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
76+
# - { os: "ubuntu-22.04", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
77+
# - { os: "windows-2022", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
8678
timeout-minutes: 50
8779
env:
8880
PACKAGE_NAME: ${{ matrix.pkg-name }}
89-
TORCH_URL: "https://download.pytorch.org/whl/cpu/torch_stable.html"
90-
TORCH_URL_STABLE: "https://download.pytorch.org/whl/cpu/torch_stable.html"
91-
TORCH_URL_TEST: "https://download.pytorch.org/whl/test/cpu/torch"
81+
TORCH_URL: "https://download.pytorch.org/whl/cpu/"
82+
TORCH_URL_STABLE: "https://download.pytorch.org/whl/cpu/"
83+
TORCH_URL_TEST: "https://download.pytorch.org/whl/test/cpu/"
9284
FREEZE_REQUIREMENTS: ${{ ! (github.ref == 'refs/heads/master' || startsWith(github.ref, 'refs/heads/release/')) }}
9385
PYPI_CACHE_DIR: "_pip-wheels"
9486
# TODO: Remove this - Enable running MPS tests on this platform
@@ -99,7 +91,7 @@ jobs:
9991
- name: Set up Python ${{ matrix.python-version }}
10092
uses: actions/setup-python@v5
10193
with:
102-
python-version: ${{ matrix.python-version }}
94+
python-version: ${{ matrix.python-version || '3.9' }}
10395

10496
- name: basic setup
10597
run: pip install -q -r .actions/requirements.txt
@@ -132,8 +124,8 @@ jobs:
132124
133125
- name: Env. variables
134126
run: |
135-
# Switch PyTorch URL
136-
python -c "print('TORCH_URL=' + str('${{env.TORCH_URL_TEST}}' if '${{ matrix.pytorch-version }}' == '2.5' else '${{env.TORCH_URL_STABLE}}'))" >> $GITHUB_ENV
127+
# Switch PyTorch URL between stable and test/future
128+
python -c "print('TORCH_URL=' + str('${{env.TORCH_URL_TEST}}' if '${{ matrix.pytorch-version }}' == '2.7' else '${{env.TORCH_URL_STABLE}}'))" >> $GITHUB_ENV
137129
# Switch coverage scope
138130
python -c "print('COVERAGE_SCOPE=' + str('lightning' if '${{matrix.pkg-name}}' == 'lightning' else 'pytorch_lightning'))" >> $GITHUB_ENV
139131
# if you install mono-package set dependency only for this subpackage
@@ -146,7 +138,7 @@ jobs:
146138
run: |
147139
pip install ".[${EXTRA_PREFIX}extra,${EXTRA_PREFIX}test,${EXTRA_PREFIX}strategies]" -U --prefer-binary \
148140
-r requirements/_integrations/accelerators.txt \
149-
--find-links="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}"
141+
--extra-index-url="${TORCH_URL}" --find-links="${PYPI_CACHE_DIR}"
150142
pip list
151143
- name: Drop LAI from extensions
152144
if: ${{ matrix.pkg-name != 'lightning' }}

‎.github/workflows/docker-build.yml

+1
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ jobs:
100100
- { python_version: "3.11", pytorch_version: "2.3.1", cuda_version: "12.1.1" }
101101
- { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.1" }
102102
- { python_version: "3.12", pytorch_version: "2.5.1", cuda_version: "12.1.1" }
103+
- { python_version: "3.12", pytorch_version: "2.6.0", cuda_version: "12.4.1" }
103104
steps:
104105
- uses: actions/checkout@v4
105106
- uses: docker/setup-buildx-action@v3

‎.github/workflows/docs-build.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ defaults:
4646

4747
env:
4848
FREEZE_REQUIREMENTS: "1"
49-
TORCH_URL: "https://download.pytorch.org/whl/cpu/torch_stable.html"
49+
TORCH_URL: "https://download.pytorch.org/whl/cpu/"
5050
PYPI_CACHE_DIR: "_pip-wheels"
5151
PYPI_LOCAL_DIR: "pypi_pkgs/"
5252

@@ -106,7 +106,7 @@ jobs:
106106
mkdir -p ${PYPI_CACHE_DIR} # in case cache was not hit
107107
ls -lh ${PYPI_CACHE_DIR}
108108
pip install .[all] -U -r requirements/${{ matrix.pkg-name }}/docs.txt \
109-
-f ${PYPI_LOCAL_DIR} -f ${PYPI_CACHE_DIR} -f ${TORCH_URL}
109+
-f ${PYPI_LOCAL_DIR} -f ${PYPI_CACHE_DIR} --extra-index-url="${TORCH_URL}"
110110
pip list
111111
- name: Install req. for Notebooks/tutorials
112112
if: matrix.pkg-name == 'pytorch'

‎.github/workflows/release-pkg.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ defaults:
2323

2424
env:
2525
FREEZE_REQUIREMENTS: 1
26-
TORCH_URL: "https://download.pytorch.org/whl/cpu/torch_stable.html"
26+
TORCH_URL: "https://download.pytorch.org/whl/cpu/"
2727
PYTHON_VER: "3.9"
2828

2929
jobs:
@@ -60,7 +60,7 @@ jobs:
6060
python-version: ${{ env.PYTHON_VER }}
6161
- name: install Package
6262
run: |
63-
pip install . -f ${TORCH_URL}
63+
pip install . --extra-index-url="${TORCH_URL}"
6464
pip list
6565
- name: package Version
6666
id: lai-package

‎dockers/base-cuda/Dockerfile

+2-3
Original file line numberDiff line numberDiff line change
@@ -92,9 +92,8 @@ RUN \
9292
-r requirements/pytorch/extra.txt \
9393
-r requirements/pytorch/test.txt \
9494
-r requirements/pytorch/strategies.txt \
95-
--find-links="https://download.pytorch.org/whl/cu${CUDA_VERSION_MM//'.'/''}/torch_stable.html" \
96-
--find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/torch" \
97-
--find-links="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/pytorch-triton"
95+
--extra-index-url="https://download.pytorch.org/whl/cu${CUDA_VERSION_MM//'.'/''}/" \
96+
--extra-index-url="https://download.pytorch.org/whl/test/cu${CUDA_VERSION_MM//'.'/''}/"
9897

9998
RUN \
10099
# Show what we have

‎requirements/fabric/base.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,5 @@
44
torch >=2.1.0, <2.6.0
55
fsspec[http] >=2022.5.0, <2024.4.0
66
packaging >=20.0, <=23.1
7-
typing-extensions >=4.4.0, <4.10.0
7+
typing-extensions >=4.4.0, <4.11.0
88
lightning-utilities >=0.10.0, <0.12.0

‎requirements/fabric/strategies.txt

+1-2
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,4 @@
66
# note: is a bug around 0.10 with `MPS_Accelerator must implement all abstract methods`
77
# shall be resolved by https://github.com/microsoft/DeepSpeed/issues/4372
88
deepspeed >=0.8.2, <=0.9.3; platform_system != "Windows" and platform_system != "Darwin" # strict
9-
bitsandbytes >=0.44.0,<0.44.2; sys_platform == 'linux' or sys_platform == 'win32'
10-
bitsandbytes >=0.42.0,<0.43.0 ; sys_platform == 'darwin'
9+
bitsandbytes >=0.45.2,<0.45.3; platform_system != "Darwin"

0 commit comments

Comments
 (0)