diff --git a/python4j/python4j-core/pom.xml b/python4j/python4j-core/pom.xml
index 1bf2b192c..4ce5a3bcd 100644
--- a/python4j/python4j-core/pom.xml
+++ b/python4j/python4j-core/pom.xml
@@ -51,4 +51,12 @@
             </exclusions>
         </dependency>
     </dependencies>
+    <profiles>
+        <profile>
+            <id>nd4j-tests-cpu</id>
+        </profile>
+        <profile>
+            <id>nd4j-tests-cuda</id>
+        </profile>
+    </profiles>
 </project>
diff --git a/python4j/python4j-numpy/pom.xml b/python4j/python4j-numpy/pom.xml
index 03e683c91..1f47ad485 100644
--- a/python4j/python4j-numpy/pom.xml
+++ b/python4j/python4j-numpy/pom.xml
@@ -56,4 +56,12 @@
             <version>${project.version}</version>
         </dependency>
     </dependencies>
+    <profiles>
+        <profile>
+            <id>nd4j-tests-cpu</id>
+        </profile>
+        <profile>
+            <id>nd4j-tests-cuda</id>
+        </profile>
+    </profiles>
 </project>