diff --git a/dockerfile b/dockerfile
index d79828ea..714203d3 100644
--- a/dockerfile
+++ b/dockerfile
@@ -2,7 +2,7 @@
 ##  Build Webserver  ##
 #######################
 
-FROM --platform=$BUILDPLATFORM mcr.microsoft.com/dotnet/sdk:9.0 AS build-Webserver
+FROM --platform=$BUILDPLATFORM mcr.microsoft.com/dotnet/sdk:9.0 AS build-webserver
 WORKDIR /WebServer
 
 # Copy the csproj
@@ -59,7 +59,7 @@ RUN wget https://www.python.org/ftp/python/3.11.0/Python-3.11.0.tgz && \
     ln -s /usr/local/bin/pip3.11 /usr/local/bin/pip
 
 RUN pip install --upgrade pip && \
-    pip install tensorflow \
+    pip install tensorflow[and-cuda] \
     fastparquet \
     openpyxl \
     yfinance \