Skip to content

Commit

Permalink
Merge branch 'release/19.21.0.20201122'
Browse files Browse the repository at this point in the history
  • Loading branch information
takuya-takeuchi committed Nov 22, 2020
2 parents cee0ba5 + eb8a716 commit ba4c35f
Show file tree
Hide file tree
Showing 56 changed files with 931 additions and 123 deletions.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,12 @@ Dlib wrapper written in C++ and C# for Windows, MacOS and Linux
|DlibDotNet for CUDA 10.2|Windows|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA102.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA102)|
||Linux|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA102.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA102)|
||OSX|-|-|-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA102.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA102)|
|DlibDotNet for CUDA 11.0|Windows|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA110.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA110)|
||Linux|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA110.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA110)|
||OSX|-|-|-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA110.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA110)|
|DlibDotNet for CUDA 11.1|Windows|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA111.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA111)|
||Linux|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA111.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA111)|
||OSX|-|-|-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.CUDA111.svg)](https://www.nuget.org/packages/DlibDotNet.CUDA111)|
|DlibDotNet for Intel MKL|Windows|||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.MKL.svg)](https://www.nuget.org/packages/DlibDotNet.MKL)|
||Linux|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.MKL.svg)](https://www.nuget.org/packages/DlibDotNet.MKL)|
||OSX|-||-|-|[![NuGet version](https://img.shields.io/nuget/v/DlibDotNet.MKL.svg)](https://www.nuget.org/packages/DlibDotNet.MKL)|
Expand Down
12 changes: 12 additions & 0 deletions docker/base/centos/7/cuda/11.0/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
FROM nvidia/cuda:11.0-cudnn8-devel-centos7
LABEL maintainer "Takuya Takeuchi <[email protected]>"

# 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
12 changes: 12 additions & 0 deletions docker/base/centos/7/cuda/11.1/Dockerfile
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 <[email protected]>"

# 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
9 changes: 9 additions & 0 deletions docker/base/ubuntu/16/cuda/11.0/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu16.04
LABEL maintainer "Takuya Takeuchi <[email protected]>"

# 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/*
9 changes: 9 additions & 0 deletions docker/base/ubuntu/16/cuda/11.1/Dockerfile
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 <[email protected]>"

# 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/*
47 changes: 47 additions & 0 deletions docker/devel/centos/7/cuda/11.0/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
FROM nvidia/cuda:11.0-cudnn8-devel-centos7
LABEL maintainer "Takuya Takeuchi <[email protected]>"

# 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
47 changes: 47 additions & 0 deletions docker/devel/centos/7/cuda/11.1/Dockerfile
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 <[email protected]>"

# 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
35 changes: 35 additions & 0 deletions docker/devel/ubuntu/16/cuda/11.0/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu16.04
LABEL maintainer "Takuya Takeuchi <[email protected]>"

# 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/*
35 changes: 35 additions & 0 deletions docker/devel/ubuntu/16/cuda/11.1/Dockerfile
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 <[email protected]>"

# 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/*
2 changes: 2 additions & 0 deletions nuget/BuildCentOS7.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@ $BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target =
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 100 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 101 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 102 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 110 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 111 }
#$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "arm"; Architecture = 64; Postfix = "64"; RID = "$OperatingSystem-arm64"; CUDA = 0 }
#$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "arm"; Architecture = 32; Postfix = ""; RID = "$OperatingSystem-arm"; CUDA = 0 }

Expand Down
2 changes: 2 additions & 0 deletions nuget/BuildUbuntu16.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@ $BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target =
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 100 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 101 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 102 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 110 }
$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "cuda"; Architecture = 64; Postfix = ""; RID = "$OperatingSystem-x64"; CUDA = 111 }
#$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "arm"; Architecture = 64; Postfix = "64"; RID = "$OperatingSystem-arm64"; CUDA = 0 }
#$BuildTargets += New-Object PSObject -Property @{ Platform = "desktop"; Target = "arm"; Architecture = 32; Postfix = ""; RID = "$OperatingSystem-arm"; CUDA = 0 }

Expand Down
14 changes: 10 additions & 4 deletions nuget/BuildUtils.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,9 @@ class Config
92,
100,
101,
102
102,
110,
111
)

$CudaVersionHash =
Expand All @@ -46,7 +48,9 @@ class Config
92 = "CUDA_PATH_V9_2";
100 = "CUDA_PATH_V10_0";
101 = "CUDA_PATH_V10_1";
102 = "CUDA_PATH_V10_2"
102 = "CUDA_PATH_V10_2";
110 = "CUDA_PATH_V11_0";
111 = "CUDA_PATH_V11_1";
}

$VisualStudio = "Visual Studio 15 2017"
Expand Down Expand Up @@ -93,7 +97,7 @@ class Config
# %4: Architecture (32/64)
# %5: Platform (desktop/android/ios/uwp)
# %6: Optional Argument
# if Target is cuda, CUDA version if Target is cuda [90/91/92/100/101/102]
# if Target is cuda, CUDA version if Target is cuda [90/91/92/100/101/102/110]
# if Target is mkl and Windows, IntelMKL directory path
#***************************************
Config( [string]$Root,
Expand Down Expand Up @@ -128,7 +132,7 @@ class Config
if ($this.PlatformArray.Contains($Platform) -eq $False)
{
$candidate = $this.PlatformArray -join "/"
Write-Host "Error: Specify Architecture [${candidate}]" -ForegroundColor Red
Write-Host "Error: Specify Platform [${candidate}]" -ForegroundColor Red
exit -1
}

Expand Down Expand Up @@ -416,6 +420,8 @@ class Config
{
# CUDA_PATH_V10_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
# CUDA_PATH_V10_1=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
# CUDA_PATH_V10_2=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
# CUDA_PATH_V11_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
# CUDA_PATH_V9_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
# CUDA_PATH_V9_1=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1
# CUDA_PATH_V9_2=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2
Expand Down
2 changes: 2 additions & 0 deletions nuget/BuildWindows.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,8 @@ $BuildTargets += New-Object PSObject -Property @{Target = "cuda"; Architecture =
$BuildTargets += New-Object PSObject -Property @{Target = "cuda"; Architecture = 64; RID = "$OperatingSystem-x64"; CUDA = 100 }
$BuildTargets += New-Object PSObject -Property @{Target = "cuda"; Architecture = 64; RID = "$OperatingSystem-x64"; CUDA = 101 }
$BuildTargets += New-Object PSObject -Property @{Target = "cuda"; Architecture = 64; RID = "$OperatingSystem-x64"; CUDA = 102 }
$BuildTargets += New-Object PSObject -Property @{Target = "cuda"; Architecture = 64; RID = "$OperatingSystem-x64"; CUDA = 110 }
$BuildTargets += New-Object PSObject -Property @{Target = "cuda"; Architecture = 64; RID = "$OperatingSystem-x64"; CUDA = 111 }

foreach ($BuildTarget in $BuildTargets)
{
Expand Down
2 changes: 2 additions & 0 deletions nuget/CreateAllPackage.bat
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@ pwsh CreatePackage.ps1 CUDA-92
pwsh CreatePackage.ps1 CUDA-100
pwsh CreatePackage.ps1 CUDA-101
pwsh CreatePackage.ps1 CUDA-102
pwsh CreatePackage.ps1 CUDA-110
pwsh CreatePackage.ps1 CUDA-111
pwsh CreatePackage.ps1 MKL
pwsh CreatePackage.ps1 UWP

Expand Down
2 changes: 2 additions & 0 deletions nuget/Publish.bat
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
dotnet nuget push DlibDotNet.MKL.%1.nupkg -k %2 -s https://api.nuget.org/v3/index.json
dotnet nuget push DlibDotNet.CUDA111.%1.nupkg -k %2 -s https://api.nuget.org/v3/index.json
dotnet nuget push DlibDotNet.CUDA110.%1.nupkg -k %2 -s https://api.nuget.org/v3/index.json
dotnet nuget push DlibDotNet.CUDA102.%1.nupkg -k %2 -s https://api.nuget.org/v3/index.json
dotnet nuget push DlibDotNet.CUDA101.%1.nupkg -k %2 -s https://api.nuget.org/v3/index.json
dotnet nuget push DlibDotNet.CUDA100.%1.nupkg -k %2 -s https://api.nuget.org/v3/index.json
Expand Down
36 changes: 33 additions & 3 deletions nuget/TestPackage.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -96,8 +96,7 @@ function RunTest($BuildTargets)
# Copy Dependencies
$OutDir = Join-Path $TargetDir bin | `
Join-Path -ChildPath Release | `
Join-Path -ChildPath netcoreapp2.0 | `
Join-Path -ChildPath $RuntimeIdentifier
Join-Path -ChildPath netcoreapp2.0
if (!(Test-Path "$OutDir")) {
New-Item "$OutDir" -ItemType Directory > $null
}
Expand All @@ -108,7 +107,8 @@ function RunTest($BuildTargets)
{
foreach($Dependency in $BuildTarget.Dependencies)
{
Copy-Item "$Dependency" "$OutDir"
$FileName = [System.IO.Path]::GetFileName("$Dependency")
New-Item -Value "$Dependency" -Path "$OutDir" -Name "$FileName" -ItemType SymbolicLink > $null
}
}
}
Expand Down Expand Up @@ -213,6 +213,34 @@ $tmp102.Add("$env:CUDA_PATH_V10_2\bin\cudnn64_7.dll")
$tmp102.Add("$env:CUDA_PATH_V10_2\bin\curand64_10.dll")
$tmp102.Add("$env:CUDA_PATH_V10_2\bin\cusolver64_10.dll")

# For DlibDotNet.CUDA110
$tmp110 = New-Object 'System.Collections.Generic.List[string]'
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cublas64_11.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cublasLt64_11.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn_adv_infer64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn_adv_train64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn_cnn_infer64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn_cnn_train64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn_ops_infer64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn_ops_train64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cudnn64_8.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\curand64_10.dll")
$tmp110.Add("$env:CUDA_PATH_V11_0\bin\cusolver64_10.dll")

# For DlibDotNet.CUDA111
$tmp111 = New-Object 'System.Collections.Generic.List[string]'
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cublas64_11.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cublasLt64_11.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn_adv_infer64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn_adv_train64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn_cnn_infer64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn_cnn_train64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn_ops_infer64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn_ops_train64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cudnn64_8.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\curand64_10.dll")
$tmp111.Add("$env:CUDA_PATH_V11_1\bin\cusolver64_11.dll")

$BuildTargets = @()
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet"; Dependencies = $null }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x86"; Architecture = 32; Package = "DlibDotNet"; Dependencies = $null }
Expand All @@ -222,6 +250,8 @@ $BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Archite
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet.CUDA100"; Dependencies = $tmp100 }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet.CUDA101"; Dependencies = $tmp101 }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet.CUDA102"; Dependencies = $tmp102 }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet.CUDA110"; Dependencies = $tmp110 }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet.CUDA111"; Dependencies = $tmp111 }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x64"; Architecture = 64; Package = "DlibDotNet.MKL"; Dependencies = $null }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "x86"; Architecture = 32; Package = "DlibDotNet.MKL"; Dependencies = $null }
$BuildTargets += New-Object PSObject -Property @{PlatformTarget = "arm"; Architecture = 32; Package = "DlibDotNet.ARM"; Dependencies = $null }
Expand Down
Loading

0 comments on commit ba4c35f

Please sign in to comment.