在 Docker 容器中使用 Python 版本 3.11 安装 PyTorch

Question

我在 PyTorch 官方页面上看到 PyTorch 支持 Python 版本 3.8 到 3.11。

当我实际尝试在 Python 3.11 Docker 映像中安装 PyTorch + CUDA 时，似乎无法找到 CUDA 驱动程序，例如

FROM python:3.11.4
RUN --mount=type=cache,id=pip-build,target=/root/.cache/pip \
    pip install torch torchaudio
ENV PATH="/usr/local/nvidia/bin:${PATH}" \
    NVIDIA_VISIBLE_DEVICES=all \
    NVIDIA_DRIVER_CAPABILITIES=all

然后，在容器内部，我看到

torch.version.cuda

是

None

比较一下

FROM pytorch/pytorch
RUN --mount=type=cache,id=pip-build,target=/root/.cache/pip \
    pip install torchaudio
ENV PATH="/usr/local/nvidia/bin:${PATH}" \
    NVIDIA_VISIBLE_DEVICES=all \
    NVIDIA_DRIVER_CAPABILITIES=all

在容器内我看到

torch.version.cuda

是

12.1

PyTorch 声称它们与 Python 3.11 兼容，有人实际上能够在 Python 3.11 中使用 PyTorch+CUDA 吗？

尝试使用 Python 3.11.4 运行 Docker 镜像

尝试运行Conda docker镜像并安装pytorch，但不断出现找不到镜像的错误

Answer 1

更新 - 我想我自己解决了它。有点愚蠢，但我只是升级了 Dockerfile 中的 Python 版本。我确信这会导致一些问题，但我还没有遇到过。

FROM pytorch/pytorch

RUN conda update -n base -c defaults conda && \
    conda install -y python=3.11 && \
    conda update --all --yes

CMD ["python", "--version"]