Skip to content

Commit 8905c8c

Browse files
committed
Mount tcpxo and fastrak libraries
1 parent f93ff24 commit 8905c8c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

runner/internal/shim/docker.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1009,6 +1009,14 @@ func configureGpus(config *container.Config, hostConfig *container.HostConfig, v
10091009
hostConfig.Mounts,
10101010
mount.Mount{Type: mount.TypeBind, Source: "/dev/aperture_devices", Target: "/dev/aperture_devices"},
10111011
)
1012+
hostConfig.Mounts = append(
1013+
hostConfig.Mounts,
1014+
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/tcpxo/lib64", Target: "/var/lib/tcpxo/lib64"},
1015+
)
1016+
hostConfig.Mounts = append(
1017+
hostConfig.Mounts,
1018+
mount.Mount{Type: mount.TypeBind, Source: "/var/lib/fastrak/lib64", Target: "/var/lib/fastrak/lib64"},
1019+
)
10121020
case host.GpuVendorAmd:
10131021
// All options are listed here: https://hub.docker.com/r/rocm/pytorch
10141022
// Only --device are mandatory, other seem to be performance-related.

0 commit comments

Comments
 (0)