1. 下载Tensorflow的Docker镜像
2. 运行镜像
docker run -it -p 8022:22 --ipc host --name docker_tensorflow --gpus all -v ~/tensorflow:/tensorflow tensorflow/tensorflow:latest-gpu /bin/bash 其中,-it为-i, -t的简写。-i为交互式,-t为开启一个终端。 -v 将主机一个目录映射到Docker容器中一个目录,如此即可以在容器中访问主机文件。
3. 打开vscode,并安装remote development插件。安装完成后,vscode左下角会有一个图标:
4. 点击该图标,此时在vscode顶部会弹出一个下拉菜单,选择其中的Remote containers: attach a running container...
5. 在弹出下拉菜单中,选择正在运行的容器,即将vscode与容器连接上。此后即可在vscode中编辑容器中的源代码文件,并进行调试了。