-
Notifications
You must be signed in to change notification settings - Fork 138
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* feat: Add LoadPng from png buffer * feat; Support CUDA 110 and 111 * feat: Update to dlib 19.21
- Loading branch information
1 parent
240b794
commit 7cac6c3
Showing
45 changed files
with
393 additions
and
47 deletions.
There are no files selected for viewing
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
FROM nvidia/cuda:11.1-cudnn8-devel-centos7 | ||
LABEL maintainer "Takuya Takeuchi <takuya.takeuchi.dev@gmail.com>" | ||
|
||
# install package to build | ||
RUN yum update -y && yum install -y \ | ||
ca-certificates | ||
RUN yum install -y https://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm | ||
RUN yum update -y && yum install -y \ | ||
libX11-devel \ | ||
lapack-devel \ | ||
openblas-devel \ | ||
&& yum clean all |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,9 @@ | ||
FROM nvidia/cuda:11.1-cudnn8-devel-ubuntu16.04 | ||
LABEL maintainer "Takuya Takeuchi <takuya.takeuchi.dev@gmail.com>" | ||
|
||
# install package to build | ||
RUN apt-get update && apt-get install -y \ | ||
libopenblas-dev \ | ||
liblapack-dev \ | ||
libx11-6 \ | ||
&& apt-get clean && rm -rf /var/lib/apt/lists/* |
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,47 @@ | ||
FROM nvidia/cuda:11.1-cudnn8-devel-centos7 | ||
LABEL maintainer "Takuya Takeuchi <takuya.takeuchi.dev@gmail.com>" | ||
|
||
# install package to build | ||
RUN yum update -y && yum install -y \ | ||
ca-certificates | ||
RUN yum install -y https://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm | ||
RUN yum update -y && yum install -y \ | ||
libX11-devel \ | ||
lapack-devel \ | ||
openblas-devel \ | ||
cmake3 | ||
RUN yum groupinstall -y "Development Tools" | ||
|
||
# set compiler | ||
ENV CMAKE_C_COMPILER=/usr/bin/gcc | ||
ENV CMAKE_CXX_COMPILER=/usr/bin/g++ | ||
|
||
# set env to build by using CUDA | ||
ENV CUDA_PATH /usr/local/cuda | ||
ENV PATH $CUDA_PATH/bin:$PATH | ||
ENV CPATH $CUDA_PATH/include:$CPATH | ||
ENV LD_LIBRARY_PATH $CUDA_PATH/lib64:$LD_LIBRARY_PATH | ||
ENV NCCL_ROOT /usr/local/nccl | ||
ENV CPATH $NCCL_ROOT/include:$CPATH | ||
ENV LD_LIBRARY_PATH $NCCL_ROOT/lib/:$LD_LIBRARY_PATH | ||
ENV LIBRARY_PATH $NCCL_ROOT/lib/:$LIBRARY_PATH | ||
|
||
# Register Microsoft key and feed | ||
RUN yum update -y && yum install -y \ | ||
curl | ||
RUN curl https://packages.microsoft.com/config/rhel/7/prod.repo > /etc/yum.repos.d/microsoft.repo | ||
RUN yum update -y && yum install -y \ | ||
powershell \ | ||
&& yum clean all | ||
|
||
# user cmake 3 instead of cmake 2 | ||
RUN alternatives --install /usr/local/bin/cmake cmake /usr/bin/cmake 10 \ | ||
--slave /usr/local/bin/ctest ctest /usr/bin/ctest \ | ||
--slave /usr/local/bin/cpack cpack /usr/bin/cpack \ | ||
--slave /usr/local/bin/ccmake ccmake /usr/bin/ccmake \ | ||
--family cmake | ||
RUN alternatives --install /usr/local/bin/cmake cmake /usr/bin/cmake3 20 \ | ||
--slave /usr/local/bin/ctest ctest /usr/bin/ctest3 \ | ||
--slave /usr/local/bin/cpack cpack /usr/bin/cpack3 \ | ||
--slave /usr/local/bin/ccmake ccmake /usr/bin/ccmake3 \ | ||
--family cmake |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,35 @@ | ||
FROM nvidia/cuda:11.1-cudnn8-devel-ubuntu16.04 | ||
LABEL maintainer "Takuya Takeuchi <takuya.takeuchi.dev@gmail.com>" | ||
|
||
# install package to build | ||
RUN apt-get update && apt-get install -y \ | ||
build-essential \ | ||
libopenblas-dev \ | ||
liblapack-dev \ | ||
libx11-dev \ | ||
cmake \ | ||
&& apt-get clean && rm -rf /var/lib/apt/lists/* | ||
|
||
# set compiler | ||
ENV CMAKE_C_COMPILER=/usr/bin/gcc | ||
ENV CMAKE_CXX_COMPILER=/usr/bin/g++ | ||
|
||
# set env to build by using CUDA | ||
ENV CUDA_PATH /usr/local/cuda | ||
ENV PATH $CUDA_PATH/bin:$PATH | ||
ENV CPATH $CUDA_PATH/include:$CPATH | ||
ENV LD_LIBRARY_PATH $CUDA_PATH/lib64:$LD_LIBRARY_PATH | ||
ENV NCCL_ROOT /usr/local/nccl | ||
ENV CPATH $NCCL_ROOT/include:$CPATH | ||
ENV LD_LIBRARY_PATH $NCCL_ROOT/lib/:$LD_LIBRARY_PATH | ||
ENV LIBRARY_PATH $NCCL_ROOT/lib/:$LIBRARY_PATH | ||
|
||
# Register Microsoft key and feed | ||
RUN apt-get update && apt-get install -y \ | ||
wget \ | ||
apt-transport-https | ||
RUN wget -q https://packages.microsoft.com/config/ubuntu/16.04/packages-microsoft-prod.deb | ||
RUN dpkg -i packages-microsoft-prod.deb && rm packages-microsoft-prod.deb | ||
RUN apt-get update && apt-get install -y \ | ||
powershell \ | ||
&& apt-get clean && rm -rf /var/lib/apt/lists/* |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.