☰

启动先前运行的具有 GPU 支持的 Docker 容器

tag-icon • tag-icon gpu docker wsl2 cuda nvidia
启动先前运行的具有 GPU 支持的 Docker 容器

我最近设置了支持 CUDA/Docker 的 WSL2（Windows 10，21H2）来训练一些神经网络。为此，我遵循了本指南中的说明：https://docs.nvidia.com/cuda/wsl-user-guide/index.html

完成后，我按照链接的文档中指定的方式运行图像：

docker run -it --gpus all -p 8888:8888 tensorflow/tensorflow:latest-gpu-py3-jupyter

此时，我可以启动浏览器并使用支持 GPU 的 Jupyter Notebook 进行工作。我通过运行以下命令来验证这一点：

import tensorflow as tf
tf.config.list_physical_devices('GPU')

输出确认 GPU 可用： [PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]。

完成后，我保存更改并停止 CTRL-C 退出容器。

到目前为止，一切都如我预期的那样进行。

为了恢复工作，我一直在使用docker start -i happy_bose我happy_bose之前运行的容器的名称。

但是，不存在 GPU 支持（调用tf.config.list_physical_devices('GPU')返回一个空列表）。

我原本希望docker start -i happy_bose启动支持 GPU 的容器，但事实并非如此。有没有办法重用支持 GPU 的容器，还是docker run每次都需要创建并启动它？

相关内容