(docker)进入容器后如何使用本机gpu

首次创建容器,不能直接使用本机gpu

系统终端进行如下配置:

1.安装NVIDIA Container Toolkit

进入Nvidia官网Installing the NVIDIA Container Toolkit — NVIDIA Container Toolkit 1.15.0 documentation,安装NVIDIA Container Toolkit

本文选用apt安装

2.进入docker文档官网,使用gpu创建并运行容器

 2.1 docker-desktop 

如果是系统安装的是docker-desktop,则进入GPU support in Docker Desktop | Docker Docs

2.2 docker engine

本文安装的是docker engine,进入

Runtime options with Memory, CPUs, and GPUs | Docker Docs

参数解释:

-it: 以交互式方式进入容器(进入容器后可以在终端输入命令)

--rm:退出容器自动删除容器(只有在测试的时候有用,一般不加)

--gpus all: 使用本机所有gpu

ubuntu : 镜像名称

nvidia-smi:执行此命令

上述命令没有指定容器名称,会自动随机生成一个名称,如果要指定名称,可使用参数

--name 容器名

例如:

docker run -it --gpus all --name bev ubuntu

以交互式方式使用ubuntu镜像创建一个名为bev的容器,可使用本机所有gpu

配置一次后,以后创建容器就可以使用本机gpu了

如果不使用本机gpu,就不加--gpus all 创建容器

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-05-16 13:42:04       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-05-16 13:42:04       100 阅读
  3. 在Django里面运行非项目文件

    2024-05-16 13:42:04       82 阅读
  4. Python语言-面向对象

    2024-05-16 13:42:04       91 阅读

热门阅读

  1. 2024年第四届长三角高校数学建模竞赛C题思路

    2024-05-16 13:42:04       35 阅读
  2. 常见递归问题

    2024-05-16 13:42:04       30 阅读
  3. css基础之定位、元素的显示与隐藏

    2024-05-16 13:42:04       31 阅读
  4. C++虚函数的使用

    2024-05-16 13:42:04       35 阅读
  5. 如何在MATALB中调用libMR

    2024-05-16 13:42:04       30 阅读
  6. 俄罗斯方块【python,小游戏】

    2024-05-16 13:42:04       28 阅读
  7. Prometheus 服务发现 添加标签

    2024-05-16 13:42:04       37 阅读
  8. Redis教程(三):Redis的五种基本数据类型

    2024-05-16 13:42:04       34 阅读
  9. Python简易 博客系统

    2024-05-16 13:42:04       31 阅读
  10. 无阻赛延时模块

    2024-05-16 13:42:04       30 阅读