在 win10 64 位上安装 CUDA 12 的 PyTorch

Question

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

在 pytorch 站点 仅显示 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 

Answer 2

$ pip3 install --pre torch==2.1.0.dev20230606+cu121 torchvision==0.16.0.dev20230606+cu121 torchaudio==2.1.0.dev20230606+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121