记录工信部医疗影像评测相关代码。
公共仓库镜像拉取
docker pull ai-hub.3incloud.com/library/pytorch/pytorch:1.2-cuda10.0-cudnn7-runtime
打包
docker run -it (image ID) /bin/bash
pip install torch1.7.0+cu101 torchvision0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
pip install albumentations opencv-contrib-python segmentation-models-pytorch
拷贝
docker cp (pwd) (container ID):/workspace/
bash文件
docker commit (container ID) ai-hub.3incloud.com/comp_18/eye
docker tag (image:1.0) ai-hub.3incloud.com/username/official
上传私有镜像仓库
docker login ai-hub.3incloud.com
docker push ai-hub.3incloud.com/comp_18/eye
沙箱
docker pull docker pull ai-hub.3incloud.com/username/classify:latest
docker run --gpus all -it -v /data/data1/眼表疾病数据:/workspace/data_eye ai-hub.3incloud.com/comp_18/eye /bin/bash
sudo docker cp (container ID):/workspace/pipeline/result.csv /data/result.csv
sftp上传
sftp [email protected]
cd sandbox_result/
put ./result.csv ./
删除镜像
docker rmi (image ID)/ ( d o c k e r i m a g e s − q ) − f 删 除 容 器 d o c k e r r m ( c o n t a i n e r I D ) / (docker images -q) -f 删除容器 docker rm (container ID)/ (dockerimages−q)−f删除容器dockerrm(containerID)/(docker ps -aq)
停止/开始/重启容器
docker stop/start/restart (container ID)
不关闭退出容器
Ctrl+P+Q
进入运行中的容器
docker attach
docker19支持运行GPU环境
安装Nvidia-container-runtime
curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
sudo apt nvidia-container-runtime
运行命令
docker run --gpus all -it -v /data/data1/眼表疾病数据:/workspace/data_eye ai-hub.3incloud.com/comp_18/eye /bin/bash
python classify_dataset_new.py --data_dir ***** --save_csv “result.csv” --device “gpu”
Nvidia-container-runtime