2025-01-17 14:38:52 +00:00
|
|
|
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-17 14:38:52 +00:00
|
|
|
|
2025-01-22 04:18:21 +00:00
|
|
|
env PATH="/opt/uv:$PATH"
|
2025-01-17 14:38:52 +00:00
|
|
|
|
2025-01-22 04:18:21 +00:00
|
|
|
arg USER_ID=1000
|
|
|
|
arg GROUP_ID=1000
|
2025-01-17 14:38:52 +00:00
|
|
|
|
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-17 14:38:52 +00:00
|
|
|
|
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-17 14:38:52 +00:00
|
|
|
|
2025-01-22 04:18:21 +00:00
|
|
|
run chown -R skynet:skynet /opt/uv
|
2025-01-17 14:38:52 +00:00
|
|
|
|
2025-01-22 04:18:21 +00:00
|
|
|
run ls /opt/uv -lah
|
2025-01-17 14:38:52 +00:00
|
|
|
|
2025-01-22 04:18:21 +00:00
|
|
|
user skynet
|
2025-01-17 14:38:52 +00:00
|
|
|
|
2025-01-22 04:18:21 +00:00
|
|
|
workdir /home/skynet
|
2025-01-17 14:38:52 +00:00
|
|
|
|
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
|
2025-01-17 14:38:52 +00:00
|
|
|
env NVIDIA_VISIBLE_DEVICES=all
|
|
|
|
|
|
|
|
copy docker/entrypoint.sh /entrypoint.sh
|
|
|
|
entrypoint ["/entrypoint.sh"]
|
|
|
|
|
|
|
|
cmd ["skynet", "--help"]
|