Skip to content

Commit 56b79f0

Browse files
committed
Fix torch and torchdata install.
1 parent 00a2884 commit 56b79f0

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

docker/Dockerfile.matgl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,5 +3,5 @@ FROM ghcr.io/astral-sh/uv:python3.12-bookworm
33
ENV DGLBACKEND=pytorch
44

55
RUN uv pip install --system dgl -f https://data.dgl.ai/wheels/torch-2.4/repo.html
6-
RUN uv pip install --system torchdata==0.8.0 torch==2.4.0
6+
RUN uv pip install --system torchdata<=0.8.0 torch<=2.4.0
77
RUN uv pip install --system matgl>=1.2.6

docker/Dockerfile.matml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,5 @@ FROM materialsvirtuallab/lammps_gnnp:latest
44
ENV DGLBACKEND=pytorch
55

66
RUN uv pip install --system dgl -f https://data.dgl.ai/wheels/torch-2.4/repo.html
7-
RUN uv pip install --system torchdata==0.8.0 torch==2.4.0
7+
RUN uv pip install --system torchdata<=0.8.0 torch<=2.4.0
88
RUN uv pip install matgl>=1.2.6 matcalc>=0.3.3 matpes --system

0 commit comments

Comments
 (0)