我想在 Maven 上有一个属性/参数,它将根据参数安装不同的依赖项。
也就是说,当用户指定时-DgpuCuda=True,对 pom.xml 的依赖会相应改变。
因此mvn -gpuCuda=True install将安装 DL4J-GPU 而不是 DL4J-CPU。
如果 -gpuCuda=True 被指定,那么这将被安装:
<dependency>
<groupId>org.nd4j</groupId>
<artifactId>nd4j-cuda-10.1</artifactId>
<version>1.0.0-beta4</version>
</dependency>
如果 -gpuCuda=False,这将被安装:
<dependency>
<groupId>org.nd4j</groupId>
<artifactId>nd4j-native</artifactId>
<version>1.0.0-beta4</version>
</dependency>
这可能吗?解决方法是什么?谢谢!!
守着一只汪
相关分类