Docker 需要将宿主机的 gpu 映射到容器内部，并且在容器内部需要调用 CUDA 库。

实现思路 {#实现思路}

1. Nvidia docker 在容器内支持 nvidia 库使用
2. 将 cuda 的库映射到容器内部
3. 在容器内部配置环境变量

操作步骤 {#操作步骤}

Nvidia docker {#Nvidia-docker}

安装方法：https://www.zywvvd.com/notes/tools/docker/nvidia-docker-install/

容器部署配置 {#容器部署配置}

使用 docker compose 部署容器

也可以指定部分 gpu 映射进容器

Cuda 库映射 {#Cuda-库映射}

在宿主机安装好 cuda 与 CuDNN 库后，在创建容器时将库路径映射进去。

需要映射 dev, cuda, gnu 文件夹：

环境变量 {#环境变量}

增加环境变量到系统：

python 中可以自动添加：

参考资料 {#参考资料}

• https://cloud.tencent.com/developer/article/1822897

• https://www.zywvvd.com/notes/tools/docker/nvidia-docker-install/

文章链接：
https://www.zywvvd.com/notes/tools/docker/docker-gpu-config/