1. 安装docker
2. 安装nvidia-container
手动下载安装包: https://github.com/NVIDIA/libnvidia-container/tree/gh-pages/stable
根据自己的服务器查看需要的版本:
在linux执行命令:
bash
hostnamectl
root@:~# hostnamectl
Static hostname: TENCENT64
Icon name: computer-server
Chassis: server
Machine ID: 859b8343e81744f5bd31debbf976fd13
Boot ID: ca0c05287c344b3e8f7d973948964c4b
Operating System: TencentOS Server 3.2 (Final)
CPE OS Name: cpe:/o:tencentos:tencentos:3
Kernel: Linux 5.4.241-1-tlinux4-0017.16
Architecture: x86-64
根据x64-64和centos定位到位置:
https://github.com/NVIDIA/libnvidia-container/tree/gh-pages/stable/centos7/x86_64
下载安装四个包:
1:nvidia-container-toolkit-base
2:libnvidia-container
3:libnvidia-container-tools
4:nvidia-container-toolkit
版本相同即可。
然后放到linux的某个文件夹,使用对应的包管理器的安装命令进行安装=
比如Centos上就是:
bash
rpm -ivh *.rpm

检查是否安装成功:
bash
nvidia-ctk --version
3. 将配置自动导入到docker
执行命令:
bash
nvidia-ctk runtime configure --runtime=docker
重新启动一下docker:
bash
systemctl restart docker
然后docker就可以使用GPU了