docker运行报错

[Jasonjk3]

C:\Windows\system32>docker run --gpus '"device=0,1'" -it -v E:\PycharmProjects\GitProject\CPM-Generate:/CPM --name=cpm dmye/cpm:v0

invalid argument "'device=0,1'" for "--gpus" flag: unexpected key ''device' in ''device=0'
See 'docker run --help'.

请问--gpus 这一部分怎么设置

[zzy14]

需要安装配置NVIDIA Docker，可以参考这个链接 https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#install-guide

[zsgd901]

这个对你会帮助
https://www.tensorflow.org/install/docker?hl=zh-cn

其中的内容为：
请通过 docker -v 检查 Docker 版本。对于 19.03 之前的版本，您需要使用 nvidia-docker2 和 --runtime=nvidia 标记；对于 19.03 及之后的版本，您将需要使用 nvidia-container-toolkit 软件包和 --gpus all 标记。这两个选项都记录在上面链接的网页上。

[Jasonjk3]

好的谢谢各位大佬，我研究下