diff --git a/Dockerfile b/Dockerfile index 39b965e..b549320 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,9 +29,8 @@ COPY requirements_base.txt /install COPY requirements_full.txt /install RUN pip install --target="/install" --upgrade -r requirements_base.txt RUN if [ "${FULL}" = "true" ]; then \ - if [ -z "{$TARGETPLATFORM}" ]; then pip install --target="/install" --upgrade torch==1.9.1+cpu torchvision==0.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html ; fi \ - ; if [ "${TARGETPLATFORM}" = "linux/amd64" ]; then pip install --target="/install" --upgrade torch==1.9.1+cpu torchvision==0.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html ; fi \ - ; if [ "${TARGETPLATFORM}" = "linux/arm64" ]; then pip install --target="/install" --upgrade torch==1.9.0 torchvision==0.10.0 -f https://torch.kmtea.eu/whl/stable.html -f https://ext.kmtea.eu/whl/stable.html ; fi \ + if [ "${TARGETPLATFORM}" = "linux/amd64" ]; then pip install --target="/install" --upgrade torch==1.13.1+cpu torchvision==0.14.1+cpu -f https://download.pytorch.org/whl/torch_stable.html ; fi \ + ; if [ "${TARGETPLATFORM}" = "linux/arm64" ]; then pip install --target="/install" --upgrade torch==1.13.1 torchvision==0.14.1 -f https://torch.kmtea.eu/whl/stable.html -f https://ext.kmtea.eu/whl/stable.html ; fi \ ; pip install --target="/install" --upgrade \ -r requirements_full.txt \ ; pip install --target="/install" --upgrade \ diff --git a/detailed_guides/INSTALLATION.md b/detailed_guides/INSTALLATION.md index c20343f..41556c0 100644 --- a/detailed_guides/INSTALLATION.md +++ b/detailed_guides/INSTALLATION.md @@ -106,7 +106,7 @@ python3.9 get-pip.py #### Install project dependencies ``` python3.9 -m pip install --ignore-installed PyYAML -python3.9 -m pip install torch==1.9.1+cpu torchvision==0.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html +python3.9 -m pip install torch==1.13.1+cpu torchvision==0.14.1+cpu -f https://download.pytorch.org/whl/torch_stable.html python3.9 -m pip install -r requirements.txt python3.9 -m pip install . ``` diff --git a/pyproject.toml b/pyproject.toml index 14995fb..515307b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -56,8 +56,8 @@ Source = "https://github.com/Kav-K/GPT3Discord" [project.optional-dependencies] full = [ - "torch==1.9.1", - "torchvision==1.10.1", + "torch==1.13.1", + "torchvision==1.14.1", "tokenizers==0.10.3", "numpy==1.24.2", "scipy==1.10.1",