Cuda presets fixes

master
agibsonccc 2021-03-29 08:15:32 +09:00
parent ac993d3e27
commit ff3162136f
1 changed files with 1 additions and 5 deletions

View File

@ -39,7 +39,7 @@
<!-- CUDA version is linked with the artifact name so cannot move to parent pom.xml --> <!-- CUDA version is linked with the artifact name so cannot move to parent pom.xml -->
<cuda.version>11.0</cuda.version> <cuda.version>11.0</cuda.version>
<cudnn.version>8.0</cudnn.version> <cudnn.version>8.0</cudnn.version>
<javacpp-presets.cuda.version>1.5.4</javacpp-presets.cuda.version> <javacpp-presets.cuda.version>1.5.5</javacpp-presets.cuda.version>
</properties> </properties>
<dependencies> <dependencies>
@ -80,14 +80,10 @@
<dependency> <dependency>
<groupId>org.junit.jupiter</groupId> <groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter-api</artifactId> <artifactId>junit-jupiter-api</artifactId>
<version>${junit.version}</version>
<scope>test</scope>
</dependency> </dependency>
<dependency> <dependency>
<groupId>org.junit.jupiter</groupId> <groupId>org.junit.jupiter</groupId>
<artifactId>junit-jupiter-engine</artifactId> <artifactId>junit-jupiter-engine</artifactId>
<version>${junit.version}</version>
<scope>test</scope>
</dependency> </dependency>
<dependency> <dependency>
<groupId>org.nd4j</groupId> <groupId>org.nd4j</groupId>