56 lines
1.7 KiB
Docker
56 lines
1.7 KiB
Docker
|
# Copyright (c) 2020-2022, NVIDIA CORPORATION. All rights reserved.
|
||
|
|
#
|
||
|
|
# NVIDIA CORPORATION and its licensors retain all intellectual property
|
||
|
|
# and proprietary rights in and to this software, related documentation
|
||
|
|
# and any modifications thereto. Any use, reproduction, disclosure or
|
||
|
|
# distribution of this software and related documentation without an express
|
||
|
|
# license agreement from NVIDIA CORPORATION is strictly prohibited.
|
||
|
|
|
||
|
|
ARG BASE_IMAGE=nvcr.io/nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04
|
||
|
|
FROM $BASE_IMAGE
|
||
|
|
|
||
|
|
RUN apt-get update -yq --fix-missing \
|
||
|
|
&& DEBIAN_FRONTEND=noninteractive apt-get install -yq --no-install-recommends \
|
||
|
|
pkg-config \
|
||
|
|
wget \
|
||
|
|
cmake \
|
||
|
|
curl \
|
||
|
|
git \
|
||
|
|
vim
|
||
|
|
|
||
|
|
#ENV PYTHONDONTWRITEBYTECODE=1
|
||
|
|
#ENV PYTHONUNBUFFERED=1
|
||
|
|
|
||
|
|
# nvidia-container-runtime
|
||
|
|
#ENV NVIDIA_VISIBLE_DEVICES all
|
||
|
|
#ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,graphics
|
||
|
|
|
||
|
|
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
|
||
|
|
RUN sh Miniconda3-latest-Linux-x86_64.sh -b -u -p ~/miniconda3
|
||
|
|
RUN ~/miniconda3/bin/conda init
|
||
|
|
RUN source ~/.bashrc
|
||
|
|
RUN conda create -n nerfstream python=3.10
|
||
|
|
RUN conda activate nerfstream
|
||
|
|
|
||
|
|
RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
|
||
|
|
# install depend
|
||
|
|
RUN conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
|
||
|
|
Copy requirements.txt ./
|
||
|
|
RUN pip install -r requirements.txt
|
||
|
|
|
||
|
|
# additional libraries
|
||
|
|
RUN pip install "git+https://github.com/facebookresearch/pytorch3d.git"
|
||
|
|
RUN pip install tensorflow-gpu==2.8.0
|
||
|
|
|
||
|
|
RUN pip uninstall protobuf
|
||
|
|
RUN pip install protobuf==3.20.1
|
||
|
|
|
||
|
|
RUN conda install ffmpeg
|
||
|
|
Copy ../python_rtmpstream /python_rtmpstream
|
||
|
|
WORKDIR /python_rtmpstream/python
|
||
|
|
RUN pip install .
|
||
|
|
|
||
|
|
Copy ../nerfstream /nerfstream
|
||
|
|
WORKDIR /nerfstream
|
||
|
|
CMD ["python3", "app.py"]
|