skynet/docker/Dockerfile.runtime+cuda-py312

44 lines
864 B
Docker
Raw Permalink Normal View History

from nvidia/cuda:12.4.1-devel-ubuntu22.04
env DEBIAN_FRONTEND=noninteractive
run apt-get update && apt-get install -y \
git \
2025-01-22 04:18:21 +00:00
curl \
libgl1 \
libglib2.0-0 \
libglu1-mesa \
libgl1-mesa-glx
2025-01-22 04:18:21 +00:00
env PATH="/opt/uv:$PATH"
2025-01-22 04:18:21 +00:00
arg USER_ID=1000
arg GROUP_ID=1000
2025-01-22 04:18:21 +00:00
run groupadd -g $GROUP_ID skynet \
&& useradd -l -u $USER_ID -g skynet -s /bin/bash skynet \
&& mkdir -p /home/skynet \
&& chown -R skynet:skynet /home/skynet
2025-01-22 04:18:21 +00:00
run curl -LsSf https://astral.sh/uv/install.sh | env UV_UNMANAGED_INSTALL="/opt/uv" sh
2025-01-22 04:18:21 +00:00
run chown -R skynet:skynet /opt/uv
2025-01-22 04:18:21 +00:00
run ls /opt/uv -lah
2025-01-22 04:18:21 +00:00
user skynet
2025-01-22 04:18:21 +00:00
workdir /home/skynet
2025-01-22 04:18:21 +00:00
run uv venv --python 3.12
workdir /home/skynet/target
env PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128
env NVIDIA_VISIBLE_DEVICES=all
copy docker/entrypoint.sh /entrypoint.sh
entrypoint ["/entrypoint.sh"]
cmd ["skynet", "--help"]