Merge pull request #181 from hodanov/feature/dockerfile

Update pytorch.
2025-02-08 23:10:52 +09:00
parent 0b3acdaea8 f3221b3541
commit b9559d2415
1 changed files with 1 additions and 1 deletions
@@ -5,4 +5,4 @@ RUN apt-get update \
    && apt-get autoremove -y \
    && apt-get clean -y \
    && rm -rf /var/lib/apt/lists/* \
-    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121 --no-cache-dir
+    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126 --no-cache-dir