626_privacy/dawnbench_jax
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add dockerfile

Browse source
This commit is contained in:
Akemi Izuko 2024-11-11 23:54:14 -07:00
parent 3fca3d08d5
commit 57d192d62b
Signed by: akemi
GPG key ID: 8DE0764E1809E9FC
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
Dockerfile Normal file
View 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"]