1

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

RuntimeError: inverse: LAPACK library not found in compilation

重现它的最简单方法是：

import torch
A = torch.rand(5,5)
torch.inverse(A)

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

#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

我不是很有经验，所以我不知道是否需要提供更多详细信息。如果是，请告诉我。

4

1 回答 1

1

我解决了我自己的问题。在火炬编译之前，我在 dockerfile 上添加了 apt-get liblapack-dev 。然后我再次运行 docker 容器并且它工作了。

于 2021-07-25T15:47:29.490 回答