Upgrade to pytorch==2.2.0 (#217)

predibase · Feb 1, 2024 · 56dc6e2 · 56dc6e2
1 parent 3b4c973
commit 56dc6e2
Show file tree

Hide file tree

Showing 5 changed files with 80 additions and 71 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -37,7 +37,7 @@ RUN cargo build --release
 # Adapted from: https://github.com/pytorch/pytorch/blob/master/Dockerfile
 FROM debian:bullseye-slim as pytorch-install
 
-ARG PYTORCH_VERSION=2.1.2
+ARG PYTORCH_VERSION=2.2.0
 ARG PYTHON_VERSION=3.10
 ARG CUDA_VERSION=11.8
 ARG MAMBA_VERSION=23.1.0-1

diff --git a/server/Makefile b/server/Makefile
@@ -18,7 +18,7 @@ gen-server:
 
 install: gen-server
 	pip install pip --upgrade
-	pip install torch==2.1.2
+	pip install torch==2.2.0
 	pip install -r requirements.txt
 	pip install -e ".[bnb, accelerate, quantize, peft]"