docer环境配置

shukun

sm86降低适配版本：export TORCH_CUDA_ARCH_LIST="7.5"
提交docker image，control + D：docker commit -m "zsh" -a "zsh" d6948dd738fe zsh_o/pytorch:pytorch-1.7.1-py38-jupyter

运行docker

sudo docker run -it -v /home/zsh_o/software:/software -v /data0:/data0 -v /data1:/data1 -p 55555:55555 -p 55554:22 --gpus=all zsh_o/pytorch:pytorch-1.7.1-py38-jupyter bash /software/run.sh

run.sh

#!/bin/bash
# sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config
service ssh restart
cd /
nohup jupyter notebook --allow-root --no-browser --port 55555 --ip='*' --NotebookApp.notebook_dir=/ > jupyter.log 2>&1 &
bash