CentOS7安装nvidia-docker

本文档详细介绍了在CentOS7上安装nvidia-docker的步骤,包括检查依赖条件、安装过程、启动服务以及如何验证和使用nvidia-docker,特别是对于Tensorflow大于1.4.0版本需要CUDA 9.0以上的环境。

依赖条件

如果使用的 Tensorflow 版本大于 1.4.0,要求 CUDA 9.0 以上版本。

安装 nvidia-docker

下载安装包:

$ wget https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker-1.0.1-1.x86_64.rpm

安装:

$ rpm -ivh nvidia-docker-1.0.1-1.x86_64.rpm

启动 nvidia-docker 服务

$ sudo systemctl restart nvidia-docker

执行以下命令,若结果显示 active(running) 则说明启动成功:

$ systemctl status nvidia-docker.service
...
Active: active (running) since Thu 2018-05-10 14:12:25 CST; 5s ago 
...

使用 nvidia-docker

查看 GPU 信息:

$ nvidia-docker run --rm nvidia/cuda nvidia-smi

启动 Tensorflow:

$ docker pull tensorflow/tensorflow:1.8.0-gpu-py3
$ docker tag tensorflow/tensorflow:1.8.0-gpu-py3 tensorflow:1.8.0-gpu
$ nvidia-docker
要在CentOS 7安装nvidia-docker,可以按照以下步骤进行操作: 1. 首先,设置nvidia-docker的存储库和GPG密钥。可以使用以下命令将存储库添加到系统中: ``` distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo ``` 2. 接下来,更新包列表并安装nvidia-docker及其依赖项。可以使用以下命令完成此操作: ``` sudo yum clean expire-cache sudo yum install -y nvidia-docker2 ``` 3. 安装完成后,重新启动docker服务以使更改生效: ``` sudo systemctl restart docker ``` 4. 最后,您可以使用以下命令来测试nvidia-docker是否安装成功: ``` sudo nvidia-docker run --rm --gpus all nvidia/cuda:11.2-base nvidia-smi ``` 请注意,这些步骤是基于CentOS 7安装过程。如果您需要其他操作系统的安装指南,可以参考Docker官方文档\[3\]中提供的链接。 参考资料: \[2\] Linux安装docker \[3\] Docker官方文档 #### 引用[.reference_title] - *1* *2* [CentOS7服务器部署nvidia-docker](https://blog.youkuaiyun.com/qq_29183811/article/details/123558987)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [Ubuntu16(Centos7) dockernvidia-docker安装及使用](https://blog.youkuaiyun.com/mzpmzk/article/details/89097489)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值