我已经安装了cuda 11.8和cudnn 8.9.6，python 3.10，我需要安装哪些pytorch版本才能使用gpu

Question

pip3 安装 torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 正在使用此命令下载 pytorch，但 pytorch 未与 GPU 连接。

我已经在虚拟环境中工作，pytorch正在连接到gpu。

NVIDIA (R) Cuda 编译器驱动程序 版权所有 (c) 2005-2022 NVIDIA 公司建于 Wed_Sep_21_10:41:10_Pacific_Daylight_Time_2022 Cuda编译工具，版本11.8，V11.8.89 Build cuda_11.8.r11.8/compiler.31833905_0 这是 nvcc --version

Answer 1

Python 3.8 或 Python 3.9 通常是与 CUDA 12 和 cuDNN 8 兼容的安全选择。尝试使用 python 3.8 或 3.9 代替 3.10