docker使用GPU

崔哥

发布于 2024-03-08 10:00:40

1690

发布于 2024-03-08 10:00:40

文章被收录于专栏：崔哥的专栏崔哥的专栏

只能使用支持 cuda 的 nvidia 显卡，其他不行😭

docker run

https://docs.docker.com/config/containers/resource_constraints/#gpu

--gpus all

docker run -it --rm --gpus all nvidia/cuda:12.3.1-base-ubuntu20.04 nvidia-smi

docker compose

https://docs.docker.com/compose/gpu-support/

services:
  test:
    image: nvidia/cuda:12.3.1-base-ubuntu20.04
    command: nvidia-smi
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

gpu

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度

docker使用GPU

docker使用GPU

docker run

docker compose

社区

活动

资源

关于

腾讯云开发者

热门产品

热门推荐

更多推荐