diff --git a/Dockerfile b/Dockerfile index 921d73d31..0008cc9d5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -ARG CUDA_VERSION=11.4.3 +ARG CUDA_VERSION=11.7.1 FROM nvidia/cuda:${CUDA_VERSION}-cudnn8-runtime-ubuntu20.04 as cuda_image FROM ubuntu:20.04 VOLUME /var/lib/docker diff --git a/requirements.txt b/requirements.txt index cdcb069a6..c822a6aa8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,8 +12,8 @@ more_itertools boto3==1.25.4 botocore==1.28.4 nltk==3.7 -torch==1.12.1 -torchvision==0.13.1 +torch==1.13.1 +torchvision==0.14.1 Pillow==9.3.0 numpy==1.23.4 validators==0.20.0 diff --git a/scripts/install_torch_amd.sh b/scripts/install_torch_amd.sh index 33f9ec316..6ef38b57c 100644 --- a/scripts/install_torch_amd.sh +++ b/scripts/install_torch_amd.sh @@ -7,5 +7,5 @@ if ! [[ "$TARGETPLATFORM" ]]; then fi if [[ "$TARGETPLATFORM" != "linux/arm64" ]]; then - pip3 install --no-cache-dir torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 --upgrade + pip3 install --no-cache-dir torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117 --upgrade fi \ No newline at end of file diff --git a/src/marqo/version.py b/src/marqo/version.py index 850d8532b..88e69c66d 100644 --- a/src/marqo/version.py +++ b/src/marqo/version.py @@ -1,4 +1,4 @@ -__version__ = "1.1.0" +__version__ = "1.1.0+torch1.13.1" def get_version() -> str: