mirror of https://github.com/skygpu/skynet.git
46 lines
1.0 KiB
Docker
46 lines
1.0 KiB
Docker
from nvidia/cuda:12.4.1-devel-ubuntu22.04
|
|
from python:3.12
|
|
|
|
env DEBIAN_FRONTEND=noninteractive
|
|
|
|
run apt-get update && apt-get install -y \
|
|
git \
|
|
llvm \
|
|
ffmpeg \
|
|
libsm6 \
|
|
libxext6 \
|
|
ninja-build
|
|
|
|
# env CC /usr/bin/clang
|
|
# env CXX /usr/bin/clang++
|
|
#
|
|
# # install llvm10 as required by llvm-lite
|
|
# run git clone https://github.com/llvm/llvm-project.git -b llvmorg-10.0.1
|
|
# workdir /llvm-project
|
|
# # this adds a commit from 12.0.0 that fixes build on newer compilers
|
|
# run git cherry-pick -n b498303066a63a203d24f739b2d2e0e56dca70d1
|
|
# run cmake -S llvm -B build -G Ninja -DCMAKE_BUILD_TYPE=Release
|
|
# run ninja -C build install # -j8
|
|
|
|
run curl -sSL https://install.python-poetry.org | python3 -
|
|
|
|
env PATH "/root/.local/bin:$PATH"
|
|
|
|
copy . /skynet
|
|
|
|
workdir /skynet
|
|
|
|
env POETRY_VIRTUALENVS_PATH /skynet/.venv
|
|
|
|
run poetry install --with=cuda -v
|
|
|
|
workdir /root/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"]
|