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

I have installed cuda 11.8 and cudnn 8.9.6, python 3.10, what pytorch versions i need to install to use the gpu

提问人:Aditi Singh 提问时间:11/16/2023 最后编辑:CowAditi Singh 更新时间:11/16/2023 访问量:47

问:

pip3 install 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 构建 cuda_11.8.r11.8/compiler.31833905_0这是 nvcc --version

蟒蛇 pycharm gpu

评论


答: 暂无答案