返回列表
[64GB]PyTorch2.1.0-Cann8.0_rc1.alpha002.alpha003-Python3.9-Euler2.8.3
PyTorch
下载量:240
上传者:duanxinyu
0
8
收藏
版本信息
Python版本:3.9
CANN版本:8.0_rc1.alpha002
操作系统版本:Euler2.8
可在MA和BMS上使用
CANN版本:8.0_rc1.alpha002
操作系统版本:Euler2.8
可在MA和BMS上使用
使用说明
# 启动指令示例
~~~
docker run -it -u root \
--device=/dev/davinci0 \
--device=/dev/davinci1 \
--device=/dev/davinci2 \
--device=/dev/davinci3 \
--device=/dev/davinci4 \
--device=/dev/davinci5 \
--device=/dev/davinci6 \
--device=/dev/davinci7 \
--device=/dev/davinci_manager \
--device=/dev/devmm_svm \
--device=/dev/hisi_hdc \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/firmware:/usr/local/Ascend/firmware \
-v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
-v /home/aicc:/home/ma-user/work/aicc \
--name pytorch_ma \
--entrypoint=/bin/bash \
swr.cn-central-221.ovaijisuan.com/dxy/pytorch_2_1_0:PyTorch2.1.0-cann8.0_rc1.alpha002_py_3.9-euler_2.8.3-64GB
~~~
# dockerfile示例
~~~
FROM registry-cbu.huawei.com/modelarts-images-manager/cache/euleros_arm:2.8.3
COPY Ascend-cann-toolkit_7.0.0_linux-aarch64.run /root
COPY Ascend-hdk-*-npu-driver_23.0.rc2.2_linux-aarch64.run /root
COPY Ascend-cann-kernels-*_7.0.0_linux.run /root
COPY Miniforge3-Linux-aarch64.sh /root
COPY requirements.txt /root
USER root
RUN cd /root && \
chmod +x A*.run && \
./Ascend-hdk-*-npu-driver_23.0.rc2.2_linux-aarch64.run --quiet --docker && \
./Ascend-cann-toolkit_7.0.0_linux-aarch64.run --quiet --full && \
./Ascend-cann-kernels-*_7.0.0_linux.run --quiet --install && \
chmod 777 Miniforge3-Linux-aarch64.sh && \
mv Miniforge3-Linux-aarch64.sh /home/ma-user && \
chmod 777 requirements.txt && \
mv requirements.txt /home/ma-user && \
rm -rf A*.run && \
rm -rf /tmp/*
ENV LD_LIBRARY_PATH=/usr/lib/aarch64-linux-gnu/hdf5/serial:/usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/driver:/usr/local/Ascend/ascend-toolkit/latest/lib64:/usr/local/Ascend/ascend-toolkit/latest/compiler/lib64/plugin/opskernel:/usr/local/Ascend/ascend-toolkit/latest/compiler/lib64/plugin/nnengine:$LD_LIBRARY_PATH \
PYTHONPATH=/usr/local/Ascend/ascend-toolkit/latest/python/site-packages:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe:$PYTHONPATH \
PATH=/usr/local/Ascend/ascend-toolkit/latest/bin:/usr/local/Ascend/ascend-toolkit/latest/compiler/ccec_compiler/bin:$PATH \
ASCEND_AICPU_PATH=/usr/local/Ascend/ascend-toolkit/latest \
ASCEND_OPP_PATH=/usr/local/Ascend/ascend-toolkit/latest/opp \
TOOLCHAIN_HOME=/usr/local/Ascend/ascend-toolkit/latest/toolkit \
ASCEND_AUTOML_PATH=/usr/local/Ascend/ascend-toolkit/latest/tools
ENV PYTHONPATH=/usr/local/Ascend/tfplugin/latest/python/site-packages:$PYTHONPATH
USER ma-user
RUN bash /home/ma-user/bind/Miniforge3-Linux-aarch64.sh -ub -p /home/ma-user/anaconda3 && \
source ~/.bashrc && \
. /home/ma-user/anaconda3/etc/profile.d/conda.sh && \
conda create -y --name MindSpore python=3.9 && \
conda activate MindSpore && \
pip --default-timeout=1000 install /usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64/te-*-py3-none-any.whl && \
pip --default-timeout=1000 install /usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64/hccl-*-py3-none-any.whl && \
pip --default-timeout=1000 install /usr/local/Ascend/ascend-toolkit/latest/tools/hccl_parser-*-py3-none-any.whl && \
pip install torch-2.1*.whl && \
pip install torch_npu*.whl && \
pip install -r requirements.txt
WORKDIR /home/ma-user
~~~
镜像地址
在其他AICC使用本镜像时,
1) 在本地arm宿主机通过docker pull 拉取上面镜像到本地(即执行docker pull remote_image_address)
2) 用docker tag 将局点信息和组织名替换成对应版本(即执行 docker tag local_image_address remote_image_address),
3) 用docker push 将修改后的镜像名称推送到局点的swr服务中(即执行docker push remote_image_address)
1) 在本地arm宿主机通过docker pull 拉取上面镜像到本地(即执行docker pull remote_image_address)
2) 用docker tag 将局点信息和组织名替换成对应版本(即执行 docker tag local_image_address remote_image_address),
3) 用docker push 将修改后的镜像名称推送到局点的swr服务中(即执行docker push remote_image_address)
swr.cn-central-221.ovaijisuan.com/dxy/pytorch_2_1_0:PyTorch2.1.0-cann8.0_rc1.alpha002_py_3.9-euler_2.8.3-64GB
评论
若您有任何疑虑,您可以在这里反馈您的建议。
发布
Error response from daemon: manifest for swr.cn-central-221.ovaijisuan.com/dxy/pytorch_2_1_0:PyTorch2.1.0-cann8.0_rc1.alpha002_py_3.9-euler_2.8.3-64GB not found: manifest unknown: Not Found tag
2024-07-19 09:29
发布
Error response from daemon: manifest for swr.cn-central-221.ovaijisuan.com/dxy/pytorch_2_1_0:PyTorch2.1.0-cann8.0_rc1.alpha002_py_3.9-euler_2.8.3-64GB not found
希望尽快修复
希望尽快修复
2024-05-31 17:26
发布