From b3621e1f3ac3aaa9f562b1afc695652c75a614ea Mon Sep 17 00:00:00 2001
From: Zhiyuan He <362583303@qq.com>
Date: Thu, 11 Mar 2021 10:27:30 +0800
Subject: [PATCH] [Cherry-pick] Update doc about nvidia-docker2 (#5366)
---
docs/manual/cluster-admin/installation-guide.md | 2 +-
docs_zh_CN/manual/cluster-admin/installation-guide.md | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/docs/manual/cluster-admin/installation-guide.md b/docs/manual/cluster-admin/installation-guide.md
index b0a839bb92..aaf1269b0b 100644
--- a/docs/manual/cluster-admin/installation-guide.md
+++ b/docs/manual/cluster-admin/installation-guide.md
@@ -115,7 +115,7 @@ Please check the following requirements for different types of worker machines:
The same as CPU worker
, and with the following additional requirements:
sudo docker run --rm nvidia/cuda:10.0-base nvidia-smi
to check it. This command should output information of available GPUs if it is setup properly. Refer to the installation guidance if it is not successfully set up.sudo docker run --rm nvidia/cuda:10.0-base nvidia-smi
to check it. This command should output information of available GPUs if it is setup properly. Refer to the installation guidance if it is not successfully set up. We don't recommend to use nvidia-docker2
. For a detailed comparison between nvidia-container-runtime
and nvidia-docker2
, please refer to here. CPU worker
一样的要求,除此之外还有下面的额外要求:
sudo docker run --rm nvidia/cuda:10.0-base nvidia-smi
来检查这一项。如果该命令成功打出当前可用的显卡个数,就说明设置是没问题的。如果它未被正确安装,请参考如何安装nvidia container runtime。sudo docker run --rm nvidia/cuda:10.0-base nvidia-smi
来检查这一项。如果该命令成功打出当前可用的显卡个数,就说明设置是没问题的。如果它未被正确安装,请参考如何安装nvidia container runtime。 我们不推荐您使用nvidia-docker2
。 有关 nvidia-container-runtime
和 nvidia-docker2
的详细对比,请参考这里。