diff --git a/Dockerfile b/Dockerfile
index 801f29d4..99fe6dde 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -57,7 +57,7 @@ COPY server/Makefile server/Makefile
 RUN cd server && make install-torch
 
 # Install specific version of transformers
-RUN cd server && make install-transformers
+RUN cd server && BUILD_EXTENSIONS="True" make install-transformers
 
 # Install server
 COPY proto proto
diff --git a/server/Makefile b/server/Makefile
index ac9cb785..6961178b 100644
--- a/server/Makefile
+++ b/server/Makefile
@@ -19,7 +19,7 @@ install-transformers:
 
 install-torch:
 	# Install specific version of torch
-	pip install torch --extra-index-url https://download.pytorch.org/whl/cu116 --no-cache-dir
+	pip install torch --extra-index-url https://download.pytorch.org/whl/cu118 --no-cache-dir
 
 install: gen-server install-torch install-transformers
 	pip install pip --upgrade