如果是跑GPU,除了检查torch和cuda的对应版本是否一致以外,还要注意官网下载的torch版本开头是cuxxx而不是cpu
pytorch各版本下载地址:https://download.pytorch.org/whl/torch_stable.html
cpu版本:

gpu版本:
确保GPU运行PyTorch时,要检查torch和cuda版本匹配,并从官方下载以cuxxx开头的版本。访问提供的链接获取不同版本的torch,注意区分CPU和GPU版本。
如果是跑GPU,除了检查torch和cuda的对应版本是否一致以外,还要注意官网下载的torch版本开头是cuxxx而不是cpu
pytorch各版本下载地址:https://download.pytorch.org/whl/torch_stable.html
cpu版本:

gpu版本:
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

被折叠的 条评论
为什么被折叠?