Win10 64位上CUDA 12的PyTorch安装

Question

我需要在我的 PC 上安装 PyTorch，其 CUDA 版本：12.0。

pytorch 站点中pytorch 2 的表格仅显示 CUDA 11.7 CUDA 11.8 作为选项。

以前版本的 PyTorch也没有在任何地方提到 CUDA 12。

最直接的方法是什么？

我是否需要使用 NGC 容器或从源代码构建 PyTorch？

谢谢，奥马尔

Answer 1

一种选择是安装支持 Python 3.11 和 CUDA 12.1 的夜间构建：

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121