Skip to content

update GPU-docker build #1186

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: 2.5
Choose a base branch
from
Open

update GPU-docker build #1186

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
48ab34e
update GPU-docker build
JasonTryharder Feb 27, 2023
cf47f43
Modify to cache nvidia-docker in CAT artifactory
Sep 11, 2023
a9ffb22
update docker_run with local path
Oct 14, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions docker/docker_build_gpu.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,8 @@ confirm() {
}

confirm
echo 'CAT docker login'
docker login cat-docker.artifacts.cat.com

DOCKER_REPO="${DOCKER_REPO:-xilinx/}"
VERSION="${VERSION:-`cat dockerfiles/VERSION.txt`}"
Expand Down
7 changes: 4 additions & 3 deletions docker/dockerfiles/vitis-ai-gpu.Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04
FROM cat-docker.artifacts.cat.com/nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04
env DEBIAN_FRONTEND=noninteractive
SHELL ["/bin/bash", "-c"]
ENV TZ=America/Denver
Expand Down Expand Up @@ -276,6 +276,7 @@ RUN if [[ ${VAI_CONDA_CHANNEL} =~ .*"tar.gz" ]]; then \
&& conda config --env --remove-key channels \
&& python3 -m pip install --upgrade pip wheel setuptools \
&& conda config --env --append channels ${VAI_CONDA_CHANNEL} \
&& conda config --env --remove channels defaults \
&& mamba env create -f /scratch/vitis-ai-optimizer_darknet.yml \
&& mamba env create -f /scratch/vitis-ai-optimizer_pytorch.yml \
&& mamba env create -f /scratch/vitis-ai-optimizer_tensorflow.yml \
Expand All @@ -290,7 +291,7 @@ RUN if [[ ${VAI_CONDA_CHANNEL} =~ .*"tar.gz" ]]; then \
&& conda activate vitis-ai-tensorflow2 \
&& mamba install --no-update-deps -y vai_q_tensorflow2_gpu pydot pyyaml jupyter ipywidgets \
dill progressbar2 pytest scikit-learn pandas matplotlib \
pillow -c conda-forge -c defaults \
pillow -c conda-forge -c anaconda \
&& pip install -r /scratch/pip_requirements.txt \
&& pip install --ignore-installed tensorflow==2.8 keras==2.8 protobuf==3.11.* \
&& pip uninstall -y h5py \
Expand All @@ -299,7 +300,7 @@ RUN if [[ ${VAI_CONDA_CHANNEL} =~ .*"tar.gz" ]]; then \
&& rm -fr ~/.cache \
&& mkdir -p $VAI_ROOT/conda/pkgs \
&& sudo chmod 777 $VAI_ROOT/conda/pkgs \
&& sudo rm /opt/vitis_ai/conda/.condarc \
# && sudo rm /opt/vitis_ai/conda/.condarc \
&& mkdir -p $VAI_ROOT/compiler \
&& conda activate vitis-ai-pytorch \
&& sudo cp -r $CONDA_PREFIX/lib/python3.7/site-packages/vaic/arch $VAI_ROOT/compiler/arch \
Expand Down
7 changes: 4 additions & 3 deletions docker_run.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -89,14 +89,15 @@ fi

docker_run_params=$(cat <<-END
-v /dev/shm:/dev/shm \
-v /opt/xilinx/dsa:/opt/xilinx/dsa \
-v /opt/xilinx/overlaybins:/opt/xilinx/overlaybins \
-v /data2/tools/xilinx/dsa:/opt/xilinx/dsa \
-v /data2/tools/xilinx/overlaybins:/opt/xilinx/overlaybins \
-e USER=$user -e UID=$uid -e GID=$gid \
-e VERSION=$VERSION \
-v $DOCKER_RUN_DIR:/vitis_ai_home \
-v /data:/data \
-v $HERE:/workspace \
-w /workspace \
--rm \
--name VAI-2.5 \
--network=host \
${DETACHED} \
${RUN_MODE} \
Expand Down