File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -18,4 +18,12 @@ RUN pip install --no-cache-dir --progress off \
1818 pytest-xdist \
1919 ruff
2020
21+ ENV ASCEND_TOOLKIT_HOME=/usr/local/Ascend/ascend-toolkit/latest
22+ ENV LD_LIBRARY_PATH=${ASCEND_TOOLKIT_HOME}/lib64:${ASCEND_TOOLKIT_HOME}/lib64/plugin/opskernel:${ASCEND_TOOLKIT_HOME}/lib64/plugin/nnengine:${ASCEND_TOOLKIT_HOME}/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64:${LD_LIBRARY_PATH}
23+ ENV PYTHONPATH=${ASCEND_TOOLKIT_HOME}/python/site-packages:${ASCEND_TOOLKIT_HOME}/opp/built-in/op_impl/ai_core/tbe:${PYTHONPATH}
24+ ENV PATH=${ASCEND_TOOLKIT_HOME}/bin:${ASCEND_TOOLKIT_HOME}/compiler/ccec_compiler/bin:${PATH}
25+ ENV ASCEND_AICPU_PATH=${ASCEND_TOOLKIT_HOME}
26+ ENV ASCEND_OPP_PATH=${ASCEND_TOOLKIT_HOME}/opp
27+ ENV TOOLCHAIN_HOME=${ASCEND_TOOLKIT_HOME}/toolkit
28+
2129WORKDIR /workspace
You can’t perform that action at this time.
0 commit comments