在 nvidia cuda docker 中运行 python 脚本

Question

I downloaded and set up the docker https://www.tensorflow.org/install/docker and the image https://github.com/NVIDIA/nvidia-docker .

测试用例工作：

docker run --gpus all -it --rm tensorflow / tensorflow: latest-gpu \
python -c "import tensorflow as tf; print (tf.reduce_sum (tf.random.normal ([1000, 1000])))"

我应该怎么做才能使用 Visual Studio Code 编写的程序在 docker 而不是 python（本地安装）中运行？

docker run --gpus all -it --rm tensorflow / tensorflow: latest-gpu \
    python /home/something.py

它不起作用，因为我理解这是 docker 中的路径，而不是主机上的路径。 我应该用里面的程序构建一个新的 docker 映像吗？ 我想使用下载的图像而不更改它。

Answer 1

您必须在 docker 上挂载您的本地目录[视觉代码保存了您的 python 文件的位置]。 这是一个工作示例 -

docker 运行 --gpus all -v <your_directory_where_visual_code_has_saved_python_file>:/home -it --rm tensorflow/tensorflow:latest-gpu
python /home/something.py

我尝试过的工作示例-

docker run --gpus all -v /home/ubuntu/test:/home/test -it tensorflow/tensorflow:latest-gpu python /home/jitest.py

Answer 2

要让您的程序在 docker 上运行，您有很多选择：

1. 在您的 Dockerfile 中复制 docker 中的代码

2. 您可以将包含 python 代码的文件夹安装为 docker 卷 [推荐]

有关更多信息，请查看此媒体文章