ubuntu16.04配置深度学习环境

# 删除原Nvidia驱动
apt-get remove --purge nvidia*
# 删除双显卡切换工具Bumlebee
apt-get purge bumblebee* bbswitch-dkms
# 禁用Ubuntu默认内置了Nvidia的第三方开源驱动nouveau，在安装nvidia驱动的时候，会和nouveau驱动冲突
vim /etc/modprobe.d/blacklist.conf
# 添加一行: blacklist nouveau
# 使更改生效
update-initramfs -u
# 重启电脑
reboot

# 添加源
add-apt-repository ppa:graphics-drivers/ppa
# 更新源
apt-get update
# 查看Ubuntu推荐的驱动版本
ubuntu-drivers devices
# 自动安装合适驱动
ubuntu-drivers autoinstall
# 查看GPU使用情况
nvidia-smi

# 安装docker
apt-get install docker-ce
# 查看docker版本,判断是否安装上
docker -v

# 安装nvidia-docker
apt-get install nvidia-docker
# 提示安装nvidia-modprobe
apt-get install nvidia-modprobe
# 启动nvidia-docker服务
systemctl start nvidia-docker
# 查看nvidia-docker服务状态
systemctl status nvidia-docker

深度学习docker: ufoym/deepo:all-py36-jupyter

nvidia-docker run -itd --name deepo -p 51002:22 -p 51006:3306 -p 51008:8888 -p 51004:50004 -v /mnt/data-sda1/deepo:/data ufoym/deepo:all-py36-jupyter

# 运行容器
nvidia-docker start deepo
# 进入容器
nvidia-docker attach deepo

docker开启ssh

# 更新源
apt-get update
# 安装ssh
apt-get install openssh-server -y

vi /etc/ssh/sshd_config
# 把PermitRootLogin prohibit-password 注释掉
# 增加一行 PermitRootLogin yes
PermitRootLogin yes

# 重启ssh服务
service ssh restart

运行jupyter notebook

jupyter notebook --no-browser --ip=0.0.0.0 --allow-root --NotebookApp.token= --notebook-dir='/data'

# 后台运行
nohup jupyter notebook --no-browser --ip=0.0.0.0 --allow-root --NotebookApp.token= --notebook-dir='/data' &

docker其他操作

# 提交容器
nvidia-docker commit deepo mydeepo
# 删除容器
nvidia-docker rm deepo
# 一行命令删除所有容器
nvidia-docker stop $(docker ps -q) & nvidia-docker rm $(docker ps -aq)