Ubuntu安装nvidia-docker

 

如果要在容器内跑使用GPU的程序,需要安装nvidia-docker,这样可以在docker内直接调用宿主机的GPU。

1. 如果直接在容器内安装nvidia的驱动,配置cuda环境,在使用时将需要指定宿主机的GPU位置,类似于/devices/gpu0 这样的。

  nvidia-docker 安装方法按照 nvidia官方文档即可;博客可 参考

  验证可用后,可以去 docker官方 找 nvidia/cuda 相关的镜像,这些镜像像其它镜像一样用即可,里面已经安装了对应版本的cuda等内容。

 

2. 版本含义,如:10.1-cudnn7-runtime-ubuntu18.04表示:ubuntu系统库,cuda10.1,cudnn7,runtime版。

** 三个版本的nvidia-docker(三个版本的size依次增大)  ——参考

base版:只有基础的库(libcudart);

runtime版:有所有的共享库,调用即可;

devel版:即开发版,包含共享库和编译、调试工具,头文件等,有bin/目录和nvcc,libcudnn在 /usr/lib/x86_64-linux-gnu/libcudnn.so。

 

posted @   谷小雨  阅读(484)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾(3.3-3.9)
· AI 智能体引爆开源社区「GitHub 热点速览」
点击右上角即可分享
微信分享提示