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