Remove the hard coded block for feature count and disable CUDA

2026-02-17 20:22:02 -08:00
parent 7ff2f994e0
commit c0fd27da07
3 changed files with 5 additions and 2 deletions
@@ -58,8 +58,9 @@ RUN wget https://www.python.org/ftp/python/3.11.0/Python-3.11.0.tgz && \
    ln -s /usr/local/bin/python3.11 /usr/local/bin/python && \
    ln -s /usr/local/bin/pip3.11 /usr/local/bin/pip

+# Dont use CUDA for the tensorflow as the CPU is way faster for this smaller workload
 RUN pip install --upgrade pip && \
-    pip install tensorflow[and-cuda] \
+    pip install tensorflow \
    fastparquet \
    openpyxl \
    yfinance \