我需要在我的 PC 上安装 PyTorch,其 CUDA 版本:12.0。
pytorch 2 的表 https://i.stack.imgur.com/X13oS.png in In 火炬网站 https://pytorch.org/get-started/locally/仅显示 CUDA 11.7 CUDA 11.8 作为选项。
PyTorch 的早期版本 https://pytorch.org/get-started/previous-versions/也没有在任何地方提到 CUDA 12。
最直接的方法是什么?
我是否需要使用 NGC 容器或从源代码构建 PyTorch?
谢谢,
奥马尔
一种选择是安装支持 Python 3.11 和 CUDA 12.1 的夜间构建:
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)