120 lines
4.2 KiB
XML
120 lines
4.2 KiB
XML
<!--~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
~ Copyright (c) 2015-2018 Skymind, Inc.
|
|
~
|
|
~ This program and the accompanying materials are made available under the
|
|
~ terms of the Apache License, Version 2.0 which is available at
|
|
~ https://www.apache.org/licenses/LICENSE-2.0.
|
|
~
|
|
~ Unless required by applicable law or agreed to in writing, software
|
|
~ distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
~ WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
~ License for the specific language governing permissions and limitations
|
|
~ under the License.
|
|
~
|
|
~ SPDX-License-Identifier: Apache-2.0
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~-->
|
|
|
|
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
|
<modelVersion>4.0.0</modelVersion>
|
|
<artifactId>deeplearning4j-cuda-11.0</artifactId>
|
|
<name>deeplearning4j-cuda</name>
|
|
<parent>
|
|
<groupId>org.deeplearning4j</groupId>
|
|
<artifactId>deeplearning4j-parent</artifactId>
|
|
<version>1.0.0-SNAPSHOT</version>
|
|
</parent>
|
|
|
|
<properties>
|
|
<!-- CUDA version is linked with the artifact name so cannot move to parent pom.xml -->
|
|
<cuda.version>11.0</cuda.version>
|
|
<cudnn.version>8.0</cudnn.version>
|
|
<javacpp-presets.cuda.version>1.5.4</javacpp-presets.cuda.version>
|
|
</properties>
|
|
|
|
<dependencyManagement>
|
|
<dependencies>
|
|
<dependency>
|
|
<groupId>org.nd4j</groupId>
|
|
<artifactId>nd4j-api</artifactId>
|
|
<version>${nd4j.version}</version>
|
|
</dependency>
|
|
</dependencies>
|
|
</dependencyManagement>
|
|
|
|
<dependencies>
|
|
<dependency>
|
|
<groupId>org.nd4j</groupId>
|
|
<artifactId>nd4j-cuda-${cuda.version}</artifactId>
|
|
<version>${nd4j.version}</version>
|
|
</dependency>
|
|
|
|
<dependency>
|
|
<groupId>org.slf4j</groupId>
|
|
<artifactId>slf4j-api</artifactId>
|
|
</dependency>
|
|
|
|
<dependency>
|
|
<groupId>ch.qos.logback</groupId>
|
|
<artifactId>logback-classic</artifactId>
|
|
<scope>test</scope>
|
|
</dependency>
|
|
|
|
<dependency>
|
|
<groupId>org.nd4j</groupId>
|
|
<artifactId>nd4j-api</artifactId>
|
|
</dependency>
|
|
<dependency>
|
|
<groupId>org.deeplearning4j</groupId>
|
|
<artifactId>deeplearning4j-core</artifactId>
|
|
<version>${project.version}</version>
|
|
</dependency>
|
|
<dependency>
|
|
<groupId>org.bytedeco</groupId>
|
|
<artifactId>javacpp</artifactId>
|
|
<version>${javacpp.version}</version>
|
|
</dependency>
|
|
<dependency>
|
|
<groupId>org.bytedeco</groupId>
|
|
<artifactId>cuda-platform</artifactId>
|
|
<version>${cuda.version}-${cudnn.version}-${javacpp-presets.cuda.version}</version>
|
|
</dependency>
|
|
<dependency>
|
|
<groupId>junit</groupId>
|
|
<artifactId>junit</artifactId>
|
|
<scope>test</scope>
|
|
</dependency>
|
|
<dependency>
|
|
<groupId>org.deeplearning4j</groupId>
|
|
<artifactId>deeplearning4j-common-tests</artifactId>
|
|
<version>${project.version}</version>
|
|
<scope>test</scope>
|
|
</dependency>
|
|
</dependencies>
|
|
|
|
<build>
|
|
<!-- Uncomment to run the full test suite using cuDNN.
|
|
<testSourceDirectory>../deeplearning4j-core/src/test/java</testSourceDirectory>
|
|
-->
|
|
</build>
|
|
|
|
<profiles>
|
|
<profile>
|
|
<id>test-nd4j-native</id>
|
|
<build>
|
|
<plugins>
|
|
<plugin>
|
|
<artifactId>maven-surefire-plugin</artifactId>
|
|
<configuration>
|
|
<skip>true</skip>
|
|
</configuration>
|
|
</plugin>
|
|
</plugins>
|
|
</build>
|
|
</profile>
|
|
<profile>
|
|
<id>test-nd4j-cuda-11.0</id>
|
|
</profile>
|
|
</profiles>
|
|
|
|
</project>
|