-
Notifications
You must be signed in to change notification settings - Fork 1
/
build.sbt
77 lines (56 loc) · 3.09 KB
/
build.sbt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
name := "7200_project"
version := "1.0"
scalaVersion := "2.11.8"
assemblyJarName in assembly := "7200project.jar"
//mainClass in (Compile, packageBin) := Some("UNET_TRAIN_SPARK")
val dl4jVersion= "1.0.0-beta3"
val sparkVersion= "2.1.0"
libraryDependencies ++= Seq(
// https://mvnrepository.com/artifact/org.apache.spark/spark-core
"org.apache.spark" %% "spark-core" % sparkVersion % "provided",
// https://mvnrepository.com/artifact/org.apache.spark/spark-sql
// "org.apache.spark" %% "spark-sql" % sparkVersion % Provided,
// https://mvnrepository.com/artifact/org.apache.hadoop/hadoop-client
// "org.apache.hadoop" % "hadoop-client" % "2.2.0" % Provided,
// https://mvnrepository.com/artifact/org.deeplearning4j/dl4j-spark
("org.deeplearning4j" %% "dl4j-spark" % s"${dl4jVersion}_spark_2")
.exclude("org.apache.spark","spark-core"),
// .exclude("org.deeplearning4j","deeplearning4j-ui-components"),
// https://mvnrepository.com/artifact/org.deeplearning4j/dl4j-spark-parameterserver
("org.deeplearning4j" %% "dl4j-spark-parameterserver" % s"${dl4jVersion}_spark_2")
.exclude("org.apache.spark", "spark-core")
.exclude("org.deeplearning4j","dl4j-spark_2.11"),
// https://mvnrepository.com/artifact/org.deeplearning4j/deeplearning4j-core
"org.deeplearning4j" % "deeplearning4j-core" % dl4jVersion,
// https://mvnrepository.com/artifact/org.deeplearning4j/deeplearning4j-nn
// "org.deeplearning4j" % "deeplearning4j-nn" % dl4jVersion,
"org.nd4j" % "nd4j-native-platform" % dl4jVersion,
// https://mvnrepository.com/artifact/org.deeplearning4j/deeplearning4j-ui
"org.deeplearning4j" %% "deeplearning4j-ui" % dl4jVersion
)
assemblyMergeStrategy in assembly := {
// case PathList("META-INF", "maven", xs @ _*) => MergeStrategy.last
case PathList("META-INF", "MANIFEST.MF") => MergeStrategy.discard
// case x if Assembly.isConfigFile(x) => MergeStrategy.concat
// case PathList(ps @ _*) if Assembly.isReadme(ps.last) || Assembly.isLicenseFile(ps.last) => MergeStrategy.rename
// case PathList("org","nd4j","serde",xs @ _*) => MergeStrategy.rename
// case PathList("com","typesafe",xs @ _*) => MergeStrategy.first
// case PathList("org","deeplearning4j",xs @ _*) => MergeStrategy.last
// case PathList("org","nd4j",xs @ _*) => MergeStrategy.first
// case PathList("org","apache","hadoop",xs @ _*) => MergeStrategy.last
// case PathList("org","bytedeco","javacpp-presets",xs @ _*) => MergeStrategy.first
// case PathList("commons-beanutils",xs @ _*) => MergeStrategy.last
// case PathList("org","slf4j",xs @ _*) => MergeStrategy.first
// case PathList("org","jetbrains",xs @ _*) => MergeStrategy.discard
// case PathList("org","glassfish","hk2","external",xs @ _*) => MergeStrategy.first
// case PathList("org","agrona",xs @ _*) => MergeStrategy.first
// case a =>
// val oldStrategy = (assemblyMergeStrategy in assembly).value
// oldStrategy(a)
case x =>
val oldStrategy = (assemblyMergeStrategy in assembly).value
oldStrategy(x)
}
//assemblyShadeRules in assembly := Seq(
// ShadeRule.rename("com.google.common.**" -> "repackaged.com.google.common.@1").inAll
//)