实习公司的服务器是配置好docker工具的,跑代码的时候直接在docker里搭建环境比较方便,不影响大服务器的环境,也不容易出现版本冲突。主要是搭建机器学习的环境pytorch为主。搭配cuda 利用gpu训练。
Docker 使用
查看所有容器:
docker ps -a
进入容器:
Docker start ‘id’/’name’
Docker exec -it ‘id’/’name’ /bin/bash
容器内查看代码:
cd ..
cd code
ls
退出容器:
exit
删除容器:
docker stop ‘id’/’name’
docker rm ‘id’/’name’
Docker pull pytorch 镜像:
docker pull pytorch/pytorch:1.8.0-cuda11.1-cudnn8-devel
Docker生成容器配置pytorch环境,映射代码指令:
docker run --gpus all -v /home/username/code:/code -p 5300:5300 -it --name dockername pytorch/pytorch:1.8.0-cuda11.1-cudnn8-devel