Docker

[YU-SHAO-XU]

Hi,
I ran the code in (build.sh) by cmd then executor failed running and created a container by code in (exec.sh) i don't know it's function, was there any wrong that i have conducted?because I found another issue say that we had to run dockerfiles/Dockerfile!?

[mapooon]

Instead of building the docker image with build.sh, you can pull the built image directly from:
https://hub.docker.com/r/mapooon/sbi.

[YU-SHAO-XU]

I pull the built image directly by https://hub.docker.com/r/mapooon/sbi. and something wrong happen!

[YU-SHAO-XU]

Hi
i try to test CDF data accroding to the :

CUDA_VISIBLE_DEVICES=* python3 src/inference/inference_dataset.py
-w weights/sbi.tar
-d CDF)

and the result was weird,is there any wrong!?

[xuyingzhongguo]

FROM nvidia/cuda:11.4.1-runtime-ubuntu20.04

RUN apt-get -y update &&
apt-get -y install python3

RUN apt-get update && apt-get upgrade -y &&
apt-get install -y libgl1-mesa-dev &&
apt-get install -y cmake &&
apt-get -y install python3-pip &&
apt-get -y update && apt-get install -y libopencv-dev &&
apt-get install -y git

RUN pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html &&
pip3 install dlib imutils scipy pandas opencv-python tqdm pretrainedmodels imgaug efficientnet-pytorch&&
pip3 install -U retinaface-pytorch &&
apt-get clean

WORKDIR /app

Fixed these parts and it works.

[nguyenvulong]

Instead of building the docker image with build.sh, you can pull the built image directly from: https://hub.docker.com/r/mapooon/sbi.