torch-distribute-train-scala/build.sbt at master · bytedeco/torch-distribute-train-scala · GitHub

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
ThisBuild / scalaVersion := "3.8.3"

lazy val root = (project in file("."))
  .settings(
    name := "distribute-train-scala"
  )

// 仓库
resolvers ++= Seq(
  Resolver.sonatypeCentralSnapshots,
  "aliyun-snapshot" at "https://maven.aliyun.com/repository/public"
)
updateOptions := updateOptions.value.withLatestSnapshots(true)
resolvers += "aliyunmaven" at "https://maven.aliyun.com/repository/public"

libraryDependencies ++= Seq(
  // JavaCPP
  "org.bytedeco" % "javacpp" % "1.5.13",
  "org.bytedeco" % "javacpp" % "1.5.13" classifier "linux-x86_64",

  // OpenBLAS
  "org.bytedeco" % "openblas" % "0.3.31-1.5.13",
  "org.bytedeco" % "openblas" % "0.3.31-1.5.13" classifier "linux-x86_64",

  // PyTorch
  "org.bytedeco" % "pytorch" % "2.10.0-1.5.13",
  "org.bytedeco" % "pytorch" % "2.10.0-1.5.13" classifier "linux-x86_64",

  "org.bytedeco" % "pytorch-platform-gpu" % "2.10.0-1.5.13" excludeAll(
    ExclusionRule("org.bytedeco", "cuda-platform"),
    ExclusionRule("org.bytedeco", "javacpp-platform"),
    ExclusionRule("org.bytedeco", "openblas-platform")
  ),

  // CUDA
  "org.bytedeco" % "cuda" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda" % "13.1-9.19-1.5.13" classifier "linux-x86_64",

  "org.bytedeco" % "cuda-redist" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-cublas" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-cudnn" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-cusolver" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-cusparse" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-npp" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-nccl" % "13.1-9.19-1.5.13",
  "org.bytedeco" % "cuda-redist-nvcomp" % "13.1-9.19-1.5.13",

  "org.bytedeco" % "cuda-redist" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-cublas" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-cudnn" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-cusolver" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-cusparse" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-npp" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-nccl" % "13.1-9.19-1.5.13" classifier "linux-x86_64",
  "org.bytedeco" % "cuda-redist-nvcomp" % "13.1-9.19-1.5.13" classifier "linux-x86_64",

  // OpenCV
  "org.bytedeco" % "opencv" % "4.13.0-1.5.13",
  "org.bytedeco" % "opencv" % "4.13.0-1.5.13" classifier "linux-x86_64",

  // FFmpeg
  "org.bytedeco" % "ffmpeg" % "8.0.1-1.5.13",
  "org.bytedeco" % "ffmpeg" % "8.0.1-1.5.13" classifier "linux-x86_64",

  // Gson
  "com.google.code.gson" % "gson" % "2.14.0"
)