added cuda based gpu image

2021-05-26 15:08:43 +04:00
parent 9a72149a52
commit 31b9e7e934
6 changed files with 137 additions and 0 deletions
--- a/server/docker/docker.go
+++ b/server/docker/docker.go
@@ -82,6 +82,9 @@ func BuildRunContainer(NumPorts int, GPU string) (*DockerVM,error) {
 		return nil,err
 	}
 	RespDocker.ImagePath = config.DockerFile
+	if GPU == "true" {
+		RespDocker.ImagePath = "/home/asleepyguy/p2p-rendering-computation/server/docker/containers/docker-ubuntu-sshd-gpu/"
+	}

 	// Gets docker information from env variables
 	cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())