Accelerate Machine Learning Local Development and Test Workflows with Nvidia Docker

This page summarizes the projects mentioned and recommended in the original post on dev.to

Our great sponsors
  • Sonar - Write Clean Python Code. Always.
  • InfluxDB - Access the most powerful time series database as a service
  • SaaSHub - Software Alternatives and Reviews
  • docker-cuda-demo

    Dockerfile

  • openpose

    OpenPose: Real-time multi-person keypoint detection library for body, face, hands, and foot estimation

    FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 # https://hub.docker.com/r/nvidia/cuda ENV DEBIAN_FRONTEND=noninteractive # install the dependencies for building OpenPose RUN apt-get update && # The rest is ignored for brevity. RUN pip3 install --no-cache-dir # The rest is ignored for brevity. # install cmake, clone OpenPose and download models RUN wget https://cmake.org/files/v3.20/cmake-3.20.2-linux-x86_64.tar.gz && \ # The rest is ignored for brevity. WORKDIR /openpose/build RUN alias python=python3 && cmake -DBUILD_PYTHON=OFF -DWITH_GTK=OFF -DUSE_CUDNN=ON .. # Build OpenPose. Cudnn 8 causes memory issues this is why we are using base with CUDA 10 and Cudnn 7 # Fix for CUDA 10.0 and Cudnn 7 based on the post below. # https://github.com/CMU-Perceptual-Computing-Lab/openpose/issues/1753#issuecomment-792431838 RUN sed -ie 's/set(AMPERE "80 86")/#&/g' ../cmake/Cuda.cmake && \ sed -ie 's/set(AMPERE "80 86")/#&/g' ../3rdparty/caffe/cmake/Cuda.cmake && \ make -j`nproc` && \ make install WORKDIR /openpose

  • Sonar

    Write Clean Python Code. Always.. Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.

  • HR-VITON

    Official PyTorch implementation for the paper High-Resolution Virtual Try-On with Misalignment and Occlusion-Handled Conditions (ECCV 2022).

    FROM tensorflow/tensorflow:1.15.5-gpu-py3 # Handle Nvidia public key update and update repositories for Ubuntu 18.x. #https://github.com/sangyun884/HR-VITON/issues/45 # reference: https://jdhao.github.io/2022/05/05/nvidia-apt-repo-public-key-error-fix/ RUN rm /etc/apt/sources.list.d/cuda.list RUN rm /etc/apt/sources.list.d/nvidia-ml.list RUN apt-key del 7fa2af80 # Additional reference: https://gitlab.com/nvidia/container-images/cuda/-/issues/158 RUN export this_distro="$(cat /etc/os-release | grep '^ID=' | awk -F'=' '{print $2}')" \ && export this_version="$(cat /etc/os-release | grep '^VERSION_ID=' | awk -F'=' '{print $2}' | sed 's/[^0-9]*//g')" \ && apt-key adv --fetch-keys "https://developer.download.nvidia.com/compute/cuda/repos/${this_distro}${this_version}/x86_64/3bf863cc.pub" \ && apt-key adv --fetch-keys "https://developer.download.nvidia.com/compute/machine-learning/repos/${this_distro}${this_version}/x86_64/7fa2af80.pub" # get the latest version of OpenCV RUN apt-get update && \ DEBIAN_FRONTEND=noninteractive \ apt-get install -y -qq \ wget git libopencv-dev RUN python -m pip install --upgrade pip && \ pip install matplotlib opencv-python==4.5.4.60 Pillow scipy \ azure-eventhub azure-eventhub-checkpointstoreblob-aio ipykernel WORKDIR /

  • FROM tensorflow/tensorflow:1.15.5-gpu-py3 # Handle Nvidia public key update and update repositories for Ubuntu 18.x. #https://github.com/sangyun884/HR-VITON/issues/45 # reference: https://jdhao.github.io/2022/05/05/nvidia-apt-repo-public-key-error-fix/ RUN rm /etc/apt/sources.list.d/cuda.list RUN rm /etc/apt/sources.list.d/nvidia-ml.list RUN apt-key del 7fa2af80 # Additional reference: https://gitlab.com/nvidia/container-images/cuda/-/issues/158 RUN export this_distro="$(cat /etc/os-release | grep '^ID=' | awk -F'=' '{print $2}')" \ && export this_version="$(cat /etc/os-release | grep '^VERSION_ID=' | awk -F'=' '{print $2}' | sed 's/[^0-9]*//g')" \ && apt-key adv --fetch-keys "https://developer.download.nvidia.com/compute/cuda/repos/${this_distro}${this_version}/x86_64/3bf863cc.pub" \ && apt-key adv --fetch-keys "https://developer.download.nvidia.com/compute/machine-learning/repos/${this_distro}${this_version}/x86_64/7fa2af80.pub" # get the latest version of OpenCV RUN apt-get update && \ DEBIAN_FRONTEND=noninteractive \ apt-get install -y -qq \ wget git libopencv-dev RUN python -m pip install --upgrade pip && \ pip install matplotlib opencv-python==4.5.4.60 Pillow scipy \ azure-eventhub azure-eventhub-checkpointstoreblob-aio ipykernel WORKDIR /

  • CIHP_PGN

    Code repository for Part Grouping Network, ECCV 2018

    GitHub repository

  • StyleGAN-Human

    StyleGAN-Human: A Data-Centric Odyssey of Human Generation

    StyleGAN for generating human images StyleGAN Demo

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts