Skip to content

Commit c3f1535

Browse files
committed
chore: add TORCH_DEVICE_BACKEND_AUTOLOAD environment variable to Dockerfile
1 parent 7f181ba commit c3f1535

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

scripts/images/runtime/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ COPY scripts/images/runtime/start.sh /opt/runtime/start.sh
1818
ENV PYTHONPATH=/opt/runtime/datamate/
1919
ENV UV_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cpu"
2020
ENV UV_INDEX_STRATEGY=unsafe-best-match
21+
ENV TORCH_DEVICE_BACKEND_AUTOLOAD=0
2122

2223
WORKDIR /opt/runtime
2324

0 commit comments

Comments
 (0)