暑假实习的时候需要连接到公司的服务器上跑程序,但是又没权限安装需要的库,所以开始学习使用docker来运行,简直整个世界都变美好了,回学校后要试试在Pycharm上配置docker。
1. 安装docker, 显卡驱动, cuda, nvidia-docker
2. 首次运行tensorflow官方docker image
链接:https://hub.docker.com/r/tensorflow/tensorflow/
在终端运行:
sudo nvidia-docker run -v /home/chuanhao/:$(pwd) -it tensorflow/tensorflow:latest-gpu bash
上面的命令会下载tensorflow docker image,并运行一个container
- -v:指定docker与宿主机共享的文件夹
- -i:以交互模式运行容器,通常与 -t 同时使用
- -t:为容器重新分配一个伪输入终端,通常与 -i 同时使用
- bash:在容器内执行bash命令