fix the fetching GPU problem (#13889)

apache · Jan 15, 2019 · 634d527 · 634d527
1 parent bc7ea31
commit 634d527
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/scala-package/deploy/pom.xml b/scala-package/deploy/pom.xml
@@ -19,6 +19,7 @@
   </description>
 
   <properties>
+    <MXNET_DIR>${project.parent.basedir}/..</MXNET_DIR>
     <ARTIFACT_ID>mxnet-full_2.11-${platform}-${flavor}</ARTIFACT_ID>
     <revision>1.5.0-SNAPSHOT</revision>
     <repositoryId>apache.snapshots.https</repositoryId>

diff --git a/scala-package/pom.xml b/scala-package/pom.xml
@@ -131,7 +131,7 @@
                 </goals>
                 <configuration>
                   <executable>bash</executable>
-                  <commandlineArgs>-c 'mkdir -p ${project.build.directory}; if [[ $(ldd ${MXNET_DIR}/lib/libmxnet.so | grep libcuda.so | wc -l) == "0" ]]; then echo flavor=cpu &gt; ${project.build.directory}/flavor.properties; else echo flavor=gpu &gt; ${project.build.directory}/flavor.properties; fi'</commandlineArgs>
+                  <commandlineArgs>-c 'mkdir -p ${project.build.directory}; if [[ $(ldd ${MXNET_DIR}/lib/libmxnet.so | grep libcuda | wc -l) == "0" ]]; then echo flavor=cpu &gt; ${project.build.directory}/flavor.properties; else echo flavor=gpu &gt; ${project.build.directory}/flavor.properties; fi'</commandlineArgs>
                 </configuration>
               </execution>
             </executions>