Add dockerfile
This commit is contained in:
parent
3fca3d08d5
commit
57d192d62b
1 changed files with 18 additions and 0 deletions
18
Dockerfile
Normal file
18
Dockerfile
Normal file
|
@ -0,0 +1,18 @@
|
|||
FROM docker.io/python:3.12.7-bookworm
|
||||
|
||||
WORKDIR /workspace
|
||||
|
||||
RUN apt-get update && apt-get install -y \
|
||||
build-essential \
|
||||
git \
|
||||
curl \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
RUN pip install --no-cache-dir \
|
||||
jax[cuda12] \
|
||||
equinox \
|
||||
torch \
|
||||
optax \
|
||||
torchvision
|
||||
|
||||
CMD ["/usr/bin/bash"]
|
Loading…
Reference in a new issue