Compare commits

...

2 Commits

Author SHA1 Message Date
hodanov
b9559d2415
Merge pull request #181 from hodanov/feature/dockerfile
Update pytorch.
2025-02-08 23:10:52 +09:00
hodanov
f3221b3541 Update pytorch. 2025-02-08 22:55:12 +09:00

View File

@ -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