fixed issue only 1 image running in GPU

2021-06-15 15:02:23 +04:00
parent 1730a58e53
commit 02ca82084b
2 changed files with 10 additions and 1 deletions
--- a/server/docker/docker_test.go
+++ b/server/docker/docker_test.go
@@ -21,6 +21,15 @@ func TestDocker(t *testing.T) {

 }

+func TestContainerHorovod(t *testing.T) {
+	// Testing by providing the horovod cpu image
+	_,err := BuildRunContainer(2,"false","cpuhorovod")
+
+	if err != nil {
+		t.Error(err)
+	}
+}
+
 func TestViewAllContainers(t *testing.T) {
 	_,err := ViewAllContainers()