Skip to content

Commit ad08291

Browse files
committed
Fix volumes targets
1 parent d2845f0 commit ad08291

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

runner/internal/shim/docker.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1034,19 +1034,19 @@ func configureGpus(config *container.Config, hostConfig *container.HostConfig, v
10341034
)
10351035
hostConfig.Mounts = append(
10361036
hostConfig.Mounts,
1037-
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/nvidia/lib64", Target: "/var/lib/nvidia/lib64"},
1037+
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/nvidia/lib64", Target: "/usr/local/nvidia/lib64"},
10381038
)
10391039
hostConfig.Mounts = append(
10401040
hostConfig.Mounts,
1041-
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/nvidia/bin", Target: "/var/lib/nvidia/bin"},
1041+
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/nvidia/bin", Target: "/usr/local/nvidia/bin"},
10421042
)
10431043
hostConfig.Mounts = append(
10441044
hostConfig.Mounts,
1045-
mount.Mount{Type: mount.TypeBind, Source: "/run/tcpx", Target: "/run/tcpx"},
1045+
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/tcpx/lib64", Target: "/usr/local/tcpx/lib64"},
10461046
)
10471047
hostConfig.Mounts = append(
10481048
hostConfig.Mounts,
1049-
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/tcpx/lib64", Target: "/var/lib/tcpx/lib64"},
1049+
mount.Mount{Type: mount.TypeBind, Source: "/run/tcpx", Target: "/run/tcpx"},
10501050
)
10511051
case host.GpuVendorAmd:
10521052
// All options are listed here: https://hub.docker.com/r/rocm/pytorch

0 commit comments

Comments
 (0)