switch to cuda/ubuntu base for future work
This commit is contained in:
parent
f1b1dc2152
commit
b9e52a36c2
1 changed files with 2 additions and 2 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Reference in a new issue