我有Ubuntu 20.04.6 LTS与CUDA 11.4和cuDNN 8500，Python 3.8.10。当运行以下命令时（不是从docker）：

import torch
print(torch.__version__)
print(torch.backends.cudnn.version())
print(torch.cuda.is_available())

字符串
我突突道：

1.13.1+cu117
8500
True

型
我使用以下dockerfile构建docker：

FROM python:3.8.10
RUN apt-get update && apt-get upgrade -y
RUN python -m venv /venv
ENV PATH="/venv/bin:$PATH"
RUN pip install --upgrade pip
RUN pip install torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
RUN pip install notebook
EXPOSE 8873
WORKDIR /app
COPY . /app
CMD ["jupyter", "notebook", "--ip=0.0.0.0", "--port=8873", "--no-browser", "--allow-root"]

型
我使用以下命令运行docker：第一个月
并运行相同的代码：

import torch
print(torch.__version__)
print(torch.backends.cudnn.version())
print(torch.cuda.is_available())

型
但是在那里GPU未被检测到（输出）：

1.13.1+cu117
8500
False

型
缺少了什么？我如何使用python 3.8.10和使用GPU的pytorch构建docker？