@@ -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