|
|
@@ -375,6 +375,8 @@ fn build_docker_command(
|
|
|
format!("-e NCCL_IB_DISABLE={}", nccl.ib_disable),
|
|
|
format!("-e NCCL_IB_HCA={}", nccl.ib_hca),
|
|
|
format!("-e NCCL_SOCKET_IFNAME={}", nccl.socket_ifname),
|
|
|
+ format!("-e GLOO_SOCKET_IFNAME={}", nccl.socket_ifname),
|
|
|
+ format!("-e TP_SOCKET_IFNAME={}", nccl.socket_ifname),
|
|
|
format!("-e NCCL_IB_GID_INDEX={}", nccl.ib_gid_index),
|
|
|
format!("-e HF_HOME=/cache/huggingface"),
|
|
|
format!("-v {}:/cache/huggingface", cache_dir),
|