首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >pytorch编译错误:编译中找不到LAPACK库。如何解决?

pytorch编译错误:编译中找不到LAPACK库。如何解决?
EN

Stack Overflow用户
提问于 2021-07-25 18:18:50
回答 1查看 132关注 0票数 1

我已经对我遇到的这个问题感到绝望了。

代码语言:javascript
复制
RuntimeError: inverse: LAPACK library not found in compilation

重现它的最简单方法是:

代码语言:javascript
复制
import torch
A = torch.rand(5,5)
torch.inverse(A)

我在一个docker容器中运行它。dockerfile中编译pytorch的部分是:

代码语言:javascript
复制
#PyTorch
RUN pip3 install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses


ENV PYTORCH_INST_VERSION="v1.8.1"
RUN git clone --recursive --branch ${PYTORCH_INST_VERSION} https://github.com/pytorch/pytorch pytorch-src && \
    cd pytorch-src && \
    export MAX_JOBS=$((`nproc` - 2)) && \
    export TORCH_CUDA_ARCH_LIST=${CUDA_ARCH} && \
    python3 setup.py install --prefix=/opt/pytorch && \
    cp -r /opt/pytorch/lib/python3.8/site-packages/* /usr/lib/python3/dist-packages/ && \
    cd /opt && \
    rm -rf /opt/pytorch-src

我没有超级经验,所以我不知道我是否需要提供额外的细节。如果是的话,请告诉我。

EN

回答 1

Stack Overflow用户

发布于 2021-07-25 23:47:29

我解决了我自己的问题。在torch编译之前,我在dockerfile上添加了apt-get liblapack-dev。然后我再次运行docker容器,它起作用了。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68517600

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档