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 @ 2024-03-19 17:27  谷小雨  阅读(190)  评论(0编辑  收藏  举报