zoukankan      html  css  js  c++  java
  • pytorch下DeepSpeed安装

    官方github写的很清楚了 DeepSpeed_github

    ##############################################################################
    # PyTorch
    ##############################################################################
    ENV PYTORCH_VERSION=1.2.0
    ENV TORCHVISION_VERSION=0.4.0
    ENV TENSORBOARDX_VERSION=1.8
    RUN pip install torch==${PYTORCH_VERSION}
    RUN pip install torchvision==${TORCHVISION_VERSION}
    RUN pip install tensorboardX==${TENSORBOARDX_VERSION}
    
    ##############################################################################
    # DeepSpeed
    ##############################################################################
    RUN git clone https://github.com/microsoft/DeepSpeed.git ${STAGE_DIR}/DeepSpeed
    RUN cd ${STAGE_DIR}/DeepSpeed && 
        git checkout . && 
        git checkout master && 
        ./install.sh --l
    RUN rm -rf ${STAGE_DIR}/DeepSpeed
    RUN python -c "import deepspeed; print(deepspeed.__version__)"
    

    源文档是--pip_sudo 没有管理权限,改成了local安装

    安装过程较慢,请耐心等待




    安装成功显示:

  • 相关阅读:
    Redis Pipeline原理分析
    python 装饰器,传递类以及参数
    q
    Redis的EXPIRE过期机制介绍
    z
    julia .文档
    julia 安装
    python 弧度与角度互转
    python 角度和弧度转化
    机器人行业中我们常说的roll、yaw、pitch是什么?
  • 原文地址:https://www.cnblogs.com/GoubuLi/p/13533812.html
Copyright © 2011-2022 走看看