docker 运用 gpu 网络相关配置

example

sudo docker run -it --runtime=nvidia --gpus all --network=host pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime

解析

对gpu的支持

--runtime=nvidia --gpus all

网络的支持

--network=host
便于连接宿主机的代理。

参考:
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/sample-workload.html

posted @ 2024-10-19 22:54  Rainflow  阅读(28)  评论(0编辑  收藏  举报