Skip to content

Commit

Permalink
Pin pytorch-lightning version to 1.7.7 to fix crash
Browse files Browse the repository at this point in the history
Attempts to use easy_train.sh crash with this error:

AttributeError: `Trainer.root_gpu` was deprecated in
v1.6 and is no longer accessible as of v1.8. Please use
`Trainer.strategy.root_device.index` instead.

Support for Trainer.root_gpu was removed in:
Lightning-AI/pytorch-lightning#11994
  • Loading branch information
linrock committed Nov 24, 2022
1 parent 298d0d7 commit 114ffbb
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@ RUN apt install -y python3 python3-dev python3-pip
RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y cmake

RUN pip3 install --no-cache --upgrade pip setuptools
RUN --mount=type=cache,target=/root/.cache/pip \
pip3 install torch --extra-index-url https://download.pytorch.org/whl/cu117
RUN --mount=type=cache,target=/root/.cache/pip \
pip3 install python-chess==0.31.4 psutil asciimatics pytorch-lightning GPUtil cupy-cuda117
RUN --mount=type=cache,target=/root/.cache/pip pip3 install \
torch --extra-index-url https://download.pytorch.org/whl/cu117
RUN --mount=type=cache,target=/root/.cache/pip pip3 install \
python-chess==0.31.4 psutil asciimatics pytorch-lightning==1.7.7 GPUtil cupy-cuda117

RUN ln -sf python3 /usr/bin/python

Expand Down

0 comments on commit 114ffbb

Please sign in to comment.