Skip to content

Commit ddf020d

Browse files
committed
added torch pin
1 parent 471389e commit ddf020d

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

.github/workflows/notebook-test.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,10 @@ jobs:
5353
restore-keys: |
5454
${{ runner.os }}-pip-${{ env.PYTHON_VERSION }}-
5555
56+
- name: Install compatible torch and torchvision
57+
run: |
58+
python3 -m pip install torch==2.8.0
59+
5660
- name: Install dependencies
5761
run: |
5862
if [[ "${{ steps.detect_env.outputs.is_act_runner }}" == "true" ]]; then

0 commit comments

Comments
 (0)