Skip to content

Commit f94f059

Browse files
authored
downgrade torch to 2.7.1, tensorboard to 2.15, remove pathos to avoid conflict with openpi & lerobot (#1821)
* downgrade torch to 2.7.1 to avoid cluster docker job get user id issue; remove pathos to avoid conflict with lerobot * downgrade protobuf and tensorboard due to openpi
1 parent 5c9c344 commit f94f059

2 files changed

Lines changed: 150 additions & 110 deletions

File tree

pyproject.toml

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ requires = [
33
"setuptools>=42",
44
"wheel",
55
"pybind11",
6-
"tomli>=1.1; python_version < \"3.11\"",
6+
"tomli>=1.1; python_version <= \"3.11\"",
77
]
88
build-backend = "setuptools.build_meta"
99

@@ -31,22 +31,22 @@ dependencies = [
3131
"matplotlib==3.10.1",
3232
"numpy==1.26",
3333
"opencv-python==4.11.0.86",
34-
"pathos==0.3.3",
34+
# "pathos==0.3.4", pathos removed due to conflict with lerobot (from openpi)
3535
"pillow>=8",
3636
"pre-commit==3.6.2",
37-
"protobuf>=6.31.1",
37+
"protobuf<=4.23", # to be compatible with openpi/tensorboard 2.15
3838
"psutil==7.0.0",
3939
"pybind11==2.13.6",
4040
"pybullet==2.5.0",
4141
"pydocstyle==6.3.0",
4242
"pylint==2.17.7",
4343
"pyglet==1.3.2",
4444
"rectangle-packer==2.0.4",
45-
"tensorboard==2.19.0",
45+
"tensorboard==2.15.0",
4646
"threadpoolctl==3.6.0",
47-
"torch==2.8.0+cu128",
47+
"torch==2.7.1+cu128",
4848
"torchtext==0.18.0",
49-
"torchvision==0.23.0+cu128",
49+
"torchvision==0.22.1+cu128",
5050
"wheel",
5151
"yapf==0.43.0",
5252
]

0 commit comments

Comments
 (0)