Add missing tensorflow dependency in bom

Change-Id: I43ce189a69299cd5961346fabc4eed8e9a179330
deepjavalibrary · Sep 22, 2020 · e62385c · e62385c
1 parent 0cd9aec
commit e62385c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bom/build.gradle b/bom/build.gradle
@@ -25,6 +25,7 @@ dependencies {
         api "ai.djl.tensorflow:tensorflow-model-zoo:${version}"
         api "ai.djl.tensorflow:tensorflow-native-auto:${tensorflow_version}"
         api "ai.djl.tensorflow:tensorflow-native-cpu:${tensorflow_version}"
+        api "ai.djl.tensorflow:tensorflow-native-cu101:${tensorflow_version}"
         api "ai.djl.pytorch:pytorch-engine:${version}"
         api "ai.djl.pytorch:pytorch-model-zoo:${version}"
         api "ai.djl.pytorch:pytorch-native-auto:${pytorch_version}"