• 【docker搭建pytorch环境及常用指令】


    安装docker和nvidia-docker2

    对docker18.06版本,如果想要在docker容器内使用gpu,需要安装nvidia-docker。
    对于docker版本大于19,需要安装nvidia-container-runtime,在创建容器时配置参数即可使用gpu。
    pull and run docker
    创建本地挂载目录

    mkdir pytorch1.4
    下载docker镜像1.4-cuda10.1-cudnn7-devel
    PS:devel版本镜像的才有nvcc等bin文件

    docker pull pytorch/pytorch:1.4-cuda10.1-cudnn7-devel

    查看本地已有的镜像
    docker images

    创建容器
    -it表示交互式运行容器;
    –name指定容器名;
    –shm-size设置的是共享内存大小,根据需要设置,如果不设置的话,默认64M,训练深度学习模型的时候dataloader会报错提示共享内存不足;
    -p是端口映射(主机端口:容器端口),-v是地址映射(主机文件夹:容器文件夹,注意必须用绝对路径);
    镜像的RESPOSITORY:TAG;
    进入容器后启动bash

    docker19以下版本
    sudo nvidia-docker run -it --name pytorch1.4 --shm-size=“15g” -p 8080:22 -v /home/ecust/txx/docker/container/pytorch1.4:/workspace pytorch/pytorch:1.4-cuda10.1-cudnn7-devel /bin/bash

    docker19以上版本
    docker run -it --gpus all --name pytorch1.4 --shm-size=“15g” -p 8080:22 -v /home/ecust/txx/docker/container/pytorch1.4:/workspace pytorch/pytorch:1.4-cuda10.1-cudnn7-devel /bin/bash

    使用1.5不开启交互:
    docker run -it --gpus all --name pytorch_1.5_container --shm-size=“32g” -v /opt/dataanalyse:/workspace pytorch/pytorch:1.5-cuda10.1-cudnn7-devel /bin/bash

    查看已经创建的容器
    docker ps -a

    查看运行中的容器
    docker ps

    启动容器
    docker start “container id or name”

    进入容器
    docker attach “container id or name”

    退出并停止容器
    exit

    退出并继续运行容器
    Ctrl+P+Q

    运行代码使用:
    nohup python train.py > myout.file 2>&1 &

    停止容器
    docker stop “container id or name”

    删除容器
    docker rm “container id or name”

    启动docker容器:
    docker run -it -d --name 名称 image_name

    docker容器复制主机的路径到容器中:
    docker cp /opt/huawei/dataanalyse/FoodRecognize 4a48540c5e2e:/workspace

    docker搭建pytorch环境

  • 相关阅读:
    小程序开发技术框架选型
    WPF文本框中加提示语
    亚马逊第三大卖家国家巴基斯坦电商营商环境如何?且看跨境电商数字身份验证服务商ADVANCE.AI要参解析
    【面试经典150 | 数学】加一
    go语言的高级特性
    C语言快速排序
    博客系统(SSM)
    【学习笔记】深度学习入门:基于Python的理论与实现-神经网络
    Linux命令(101)之nl
    【视觉SLAM】Fast Relocalisation and Loop Closing in Keyframe-Based SLAM
  • 原文地址:https://blog.csdn.net/qq_32907491/article/details/126692118