Cuda11.1 and cv2 import fix (#108)
* docker for cuda11 * fix cv2 import Co-authored-by: Roman Suvorov <r.suvorov@samsung.com>pull/112/head
parent
d97ac962bd
commit
4934ca6f7c
4 changed files with 48 additions and 2 deletions
@ -0,0 +1,39 @@ |
|||||||
|
FROM nvidia/cuda:11.1-runtime-ubuntu18.04 |
||||||
|
|
||||||
|
RUN apt-get update && \ |
||||||
|
apt-get upgrade -y && \ |
||||||
|
apt-get install -y wget mc tmux nano build-essential rsync libgl1 |
||||||
|
|
||||||
|
ARG USERNAME=user |
||||||
|
RUN apt-get install -y sudo && \ |
||||||
|
addgroup --gid 1000 $USERNAME && \ |
||||||
|
adduser --uid 1000 --gid 1000 --disabled-password --gecos '' $USERNAME && \ |
||||||
|
adduser $USERNAME sudo && \ |
||||||
|
echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \ |
||||||
|
USER=$USERNAME && \ |
||||||
|
GROUP=$USERNAME |
||||||
|
|
||||||
|
USER $USERNAME:$USERNAME |
||||||
|
WORKDIR "/home/$USERNAME" |
||||||
|
ENV PATH="/home/$USERNAME/miniconda3/bin:/home/$USERNAME/.local/bin:${PATH}" |
||||||
|
ENV PYTHONPATH="/home/$USERNAME/project" |
||||||
|
|
||||||
|
RUN wget -O /tmp/miniconda.sh https://repo.anaconda.com/miniconda/Miniconda3-py39_4.9.2-Linux-x86_64.sh && \ |
||||||
|
echo "536817d1b14cb1ada88900f5be51ce0a5e042bae178b5550e62f61e223deae7c /tmp/miniconda.sh" > /tmp/miniconda.sh.sha256 && \ |
||||||
|
sha256sum --check --status < /tmp/miniconda.sh.sha256 && \ |
||||||
|
bash /tmp/miniconda.sh -bt -p "/home/$USERNAME/miniconda3" && \ |
||||||
|
rm /tmp/miniconda.sh && \ |
||||||
|
conda build purge && \ |
||||||
|
conda init |
||||||
|
|
||||||
|
RUN pip install -U pip |
||||||
|
RUN pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html |
||||||
|
RUN pip install numpy scipy opencv-python tensorflow joblib matplotlib pandas \ |
||||||
|
albumentations==0.5.2 pytorch-lightning==1.2.9 tabulate easydict==1.9.0 kornia==0.5.0 webdataset \ |
||||||
|
packaging gpustat tqdm pyyaml hydra-core==1.1.0.dev6 scikit-learn==0.24.2 tabulate |
||||||
|
RUN pip install scikit-image==0.17.2 |
||||||
|
|
||||||
|
ENV TORCH_HOME="/home/$USERNAME/.torch" |
||||||
|
|
||||||
|
ADD entrypoint.sh /home/$USERNAME/.local/bin/entrypoint.sh |
||||||
|
ENTRYPOINT [ "entrypoint.sh" ] |
@ -0,0 +1,5 @@ |
|||||||
|
#!/bin/bash |
||||||
|
|
||||||
|
BASEDIR="$(dirname $0)" |
||||||
|
|
||||||
|
docker build -t windj007/lama:cuda111 -f "$BASEDIR/Dockerfile-cuda111" "$BASEDIR" |
@ -1,3 +1,5 @@ |
|||||||
#!/bin/bash |
#!/bin/bash |
||||||
|
|
||||||
docker build -t windj007/lama . |
BASEDIR="$(dirname $0)" |
||||||
|
|
||||||
|
docker build -t windj007/lama -f "$BASEDIR/Dockerfile" "$BASEDIR" |
||||||
|
Loading…
Reference in new issue