nvidia-docker找不到nvidia-smi:本地最新

Question

我正在尝试运行一个具有GPU访问权限的docker容器，并且该容器在退出时不会自动删除。

我正在尝试nvidia-docker run -it -v ~/dir/to/my/data:/data nvidia-smi，但它告诉我：

Unable to find image 'nvidia-smi:latest' locally
docker: Error response from daemon: pull access denied for nvidia-smi, repository does not exist or may require 'docker login'.
See 'docker run --help'.

现在我已经检查了这个问题，并(尽管我确实认为我的问题有点不同)尝试了nvidia-docker run -it -v '~/dir/to/my/data':/data nvidia-smi，但只得到了相同的错误。

我还创建了一个docker帐户，并使用docker login登录，但这似乎对我没有任何好处。

我该如何解决这个问题？在正确的方向上提供任何帮助或推动，我们都将非常感激。

谢谢!

Answer 1

您的命令中缺少Docker镜像名称。nvidia-smi是命令名称，而不是镜像名称。在其前面添加nvidia/cuda:9.0-base。这是一个已经存在的Docker镜像。

nvidia-docker run -it -v ~/dir/to/my/data:/data nvidia/cuda:9.0-base nvidia-smi