File tree Expand file tree Collapse file tree 3 files changed +3
-6
lines changed Expand file tree Collapse file tree 3 files changed +3
-6
lines changed Original file line number Diff line number Diff line change 23
23
24
24
sudo apt-get install -y protobuf-compiler
25
25
26
- # use RC build
27
- pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu128
28
-
29
26
pip install lintrunner lintrunner-adapters
30
27
lintrunner init
31
28
Original file line number Diff line number Diff line change @@ -15,11 +15,11 @@ jobs:
15
15
- runs-on : " linux.2xlarge"
16
16
gpu-arch-type : " cpu"
17
17
gpu-arch-version : " "
18
- torch-version : " test "
18
+ torch-version : " stable "
19
19
- runs-on : " linux.g5.12xlarge.nvidia.gpu"
20
20
gpu-arch-type : " cuda"
21
21
gpu-arch-version : " 12.4"
22
- torch-version : " test "
22
+ torch-version : " stable "
23
23
- runs-on : " linux.g5.12xlarge.nvidia.gpu"
24
24
gpu-arch-type : " cuda"
25
25
gpu-arch-version : " 12.4"
Original file line number Diff line number Diff line change @@ -12,7 +12,7 @@ classifiers = [
12
12
]
13
13
dynamic = [" version" ]
14
14
dependencies = [
15
- " torch"
15
+ " torch>=2.7 "
16
16
]
17
17
18
18
[project .urls ]
You can’t perform that action at this time.
0 commit comments