docker配置gpu运行环境:linux离线安装nvidia-container,避免网络问题
1. 安装docker
2. 安装nvidia-container
手动下载安装包: https://github.com/NVIDIA/libnvidia-container/tree/gh-pages/stable
根据自己的服务器查看需要的版本:
在linux执行命令:
hostnamectlroot@:~# hostnamectlStatic hostname: TENCENT64Icon name: computer-serverChassis: serverMachine ID: 859b8343e81744f5bd31debbf976fd13Boot ID: ca0c05287c344b3e8f7d973948964c4bOperating System: TencentOS Server 3.2 (Final)CPE OS Name: cpe:/o:tencentos:tencentos:3Kernel: Linux 5.4.241-1-tlinux4-0017.16Architecture: 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上就是:
rpm -ivh *.rpm
检查是否安装成功:
nvidia-ctk --version
3. 将配置自动导入到docker
执行命令:
nvidia-ctk runtime configure --runtime=docker
重新启动一下docker:
systemctl restart docker
然后docker就可以使用GPU了