無法在 Alpine Docker 容器中導入 PyTorch

Question

下面的 MRE 應該足以解釋我想要做什么。 谷歌沒有幫助，因為它似乎是一個獨特的錯誤，這讓我感到驚訝，因為 Alpine 和 PyTorch 都相當受歡迎。

Dockerfile：

FROM nginx:stable-alpine as production
WORKDIR /app
RUN apk update && apk add --no-cache python3 bash && \
  python3 -m ensurepip && \
  rm -r /usr/lib/python*/ensurepip && \
  pip3 install --upgrade pip setuptools && \
  if [ ! -e /usr/bin/pip ]; then ln -s pip3 /usr/bin/pip ; fi && \
  if [[ ! -e /usr/bin/python ]]; then ln -sf /usr/bin/python3 /usr/bin/python; fi && \
  rm -r /root/.cache
RUN apk update && apk add gcc g++ python3-dev musl-dev
RUN pip install torch==1.10.1 --find-links https://download.pytorch.org/whl/torch_stable.html

將 bash 構建到其中：

sudo docker build -t mini .
sudo docker run --name torch-pls -it --entrypoint bash mini

啟動 Python：

bash-5.1# python
Python 3.9.5 (default, Nov 24 2021, 21:19:13) 
[GCC 10.3.1 20210424] on linux
Type "help", "copyright", "credits" or "license" for more information.

導入 PyTorch 並看到它失敗：

>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python3.9/site-packages/torch/__init__.py", line 196, in <module>
    _load_global_deps()
  File "/usr/lib/python3.9/site-packages/torch/__init__.py", line 149, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  File "/usr/lib/python3.9/ctypes/__init__.py", line 374, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: Error relocating /usr/lib/python3.9/site-packages/torch/lib/libgomp-a34b3233.so.1: pthread_attr_setaffinity_np: symbol not found

任何人都可以幫忙嗎？

Answer 1

我已經嘗試過 PyTorch 1.11.0。 這里同樣的問題。 我認為必須從 alpine 轉移到其他基本映像。 ubuntu 也許吧！