容器中使用显卡:nvidia-container-toolkit环境安装

  1. Docker
    可以使用daocloud加速下载(http://get.daocloud.io/):
curl -sSL https://get.daocloud.io/docker | sh

或参考官方安装方法https://docs.docker.com/engine/install/

  • 检查是否安装成功:
docker -v
  1. Nvidia Driver
    自动安装:
sudo ubuntu-drivers autoinstall

或者手动从官网下载安装:https://www.nvidia.com/Download/index.aspx
安装后重启

sudo reboot
  • 检查是否安装成功
nvidia-smi
  1. nvidia-container-toolkit

                
  • 3
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
CentOS是一种流行的Linux操作系统,而NVIDIA Docker是一个用于在容器运行GPU加速应用程序的工具。下面是在CentOS上安装NVIDIA Docker的步骤: 1. 首先,确保你的CentOS系统已经安装Docker。你可以通过运行以下命令来检查: ``` docker --version ``` 2. 如果你还没有安装Docker,可以按照以下步骤进行安装: - 添加Docker的官方存储库: ``` sudo yum install -y yum-utils sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo ``` - 安装Docker引擎: ``` sudo yum install -y docker-ce docker-ce-cli containerd.io ``` - 启动Docker服务并设置开机自启: ``` sudo systemctl start docker sudo systemctl enable docker ``` 3. 安装NVIDIA驱动程序。你可以通过以下命令来安装适合你的NVIDIA显卡的驱动程序: ``` sudo yum install -y nvidia-driver-latest-dkms ``` 4. 安装NVIDIA Container Toolkit。这是一个用于在Docker容器运行GPU应用程序的工具包,其包括nvidia-docker命令。 - 添加NVIDIA的官方存储库: ``` 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 ``` - 安装NVIDIA Container Toolkit: ``` sudo yum install - 重新加载Docker配置并重启Docker服务: ``` sudo systemctl restart docker ``` 5. 验证安装是否成功。你可以运行以下命令来验证nvidia-docker是否正确安装: ``` sudo docker run --gpus all nvidia/cuda:11.0-base nvidia-smi ``` 如果一切正常,你应该能够看到NVIDIA显卡的信息。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值