使用Dokcer配置Tensorflow-1.15环境
目前学术界大部分深度学习的开源代码都是基于Pytorch的,但还有少部分工作或者以前的工作是基于Tensorflow 1.x的,由于tensorflow的版本和CUDA的版本有严格的对应关系,它需要依赖很多相应版本的CUDA的C++动态库,而CUDA又和显卡型号存在对应关系,那此时在本机上直接部署tensorflow可能会破坏现有环境。所以我们可以选择使用Docker容器生成各种环境来执行代码。更棒的是VS Code的Remote-Container插件可以支持我们使用本地的Docker容器环境进行开发。
准备工作
- 首先需要在本机上安装NVIDIA的显卡驱动,剩下的CUDA库在Docker镜像中都配置好了
1. 拉取镜像
Tensorflow安装官网: https://www.tensorflow.org/install/docker?hl=zh-cn
Tensorflow官方docker hub链接: https://hub.docker.com/r/tensorflow/tensorflow/
拉取相应版本的cuda和tensorflow镜像
docker pull nvidia/cuda:10.0-devel-ubuntu18.04
docker pull tensorflow/tensorflow:1.15.3-gpu-jupyter
拉取镜像后直接运行,有一个报错
docker: Error response from daemon: Unknown runtime specified nvidia.
谷歌以后参考Stackoverflow: https://stackoverflow.com/questions/52865988/nvidia-docker-unknown-runtime-specified-nvidia
安装nvidia runtime, 并把它添加到docker runtime的配置文件
sudo apt-get install nvidia-container-runtime
sudo vim /etc/docker/daemon.json
把下面内容添加到文件中
{
"runtimes": {
"nvidia": {
"path": "nvidia-container-runtime",
"runtimeArgs": []
}
}
}
重启docker
sudo systemctl daemon-reload
sudo systemctl restart docker
2 启动容器
2.1 作为守护进程启动(启动后一直在后台运行)
docker run -itd -v /home/qyz:/home/qyz --rm --runtime=nvidia --name tensorflow tensorflow/tensorflow:1.15.3-gpu-jupyter bash
docker ps -a
docker exec -it <容器ID> or <容器 name> bash
docker attach <容器ID> or <容器 name>
docker stop <容器 ID> or <容器 name>
docker restart <容器 ID> or <容器 name>
2.2 交互式启动(启动后直接进入容器,退出后容器自动停止)
python交互式命令行启动
docker run -it --rm --runtime=nvidia tensorflow/tensorflow:1.15.3-gpu-jupyter python
bash启动并指定挂载目录
docker run -it --rm --runtime=nvidia -v /home/qyz:/home/qyz tensorflow/tensorflow:1.15.3-gpu-jupyter bash
此时就可以进入docker容器,利用其中的环境来执行本地的代码了.
2.3 docker中启动jupyter
jupyter-notebook --no-browser --ip 0.0.0.0 --port=8888 --allow-root
3. VS Code连接容器
-
安装Remote-Containers插件
-
按F1
,选择remote-containers.attachToRunningContainer
命令,然后选择本地启动的容器
-
此时再点一下VS Code的插件,可以之前安装好的Python,Jupyter插件都是灰色的,不可用,需要根据VS Code的提示在Docker容器里再安装一下Python相关的插件,就可以正常写代码了。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)