switch to cuda/ubuntu base for future work

This commit is contained in:
Ryan Voots 2024-04-23 09:15:47 -04:00
parent f1b1dc2152
commit b9e52a36c2

View file

@ -1,7 +1,7 @@
FROM python:3.10-slim
FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
RUN apt-get update && apt-get install -y \
git-lfs \
git-lfs python3 build-essential \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app