Skip to content

P
power-node
Commit 02b8af39 authored by duanjinfei's avatar duanjinfei
Browse files
Options

updage model container start gpu driver

parent 7ebfaa19
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 5 deletions
operate/docker.go
View file @ 02b8af39
...@@ -175,9 +175,6 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd ...@@ -175,9 +175,6 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd
portMap := nat.PortMap{} portMap := nat.PortMap{}
portMap = make(map[nat.Port][]nat.PortBinding, 0) portMap = make(map[nat.Port][]nat.PortBinding, 0)
portMap[nat.Port(dockerCmd.ContainerPort+"/tcp")] = portBinds portMap[nat.Port(dockerCmd.ContainerPort+"/tcp")] = portBinds
capabilities := make([][]string, 0)
capabilities = append(capabilities, []string{"gpu"})
resp, err := d.dockerClient.ContainerCreate(ctx, &container.Config{ resp, err := d.dockerClient.ContainerCreate(ctx, &container.Config{
Image: imageName, Image: imageName,
}, &container.HostConfig{ }, &container.HostConfig{
...@@ -186,9 +183,10 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd ...@@ -186,9 +183,10 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd
Resources: container.Resources{ Resources: container.Resources{
DeviceRequests: []container.DeviceRequest{ DeviceRequests: []container.DeviceRequest{
{ {
//Driver: "nvidia", Driver: "nvidia",
Count: -1, // -1 means all GPUs Count: -1, // -1 means all GPUs
Capabilities: capabilities, Capabilities: [][]string{{"gpu"}},
Options: nil,
}, },
}, },
}, },
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment