默认下载的基本都是cpu版本的,GPU要到pytorch下对应的版本,https://pytorch.org/get-started/previous-versions/ 历届版本在这个链接,注意对应的cuda,也可以看你下载pytorch大小判断是否是gpu版本,我下的gpu有1.5G,cpu版本才200多M。
torch.cuda.is_available() = False
最新推荐文章于 2024-06-12 15:05:23 发布
默认下载的基本都是cpu版本的,GPU要到pytorch下对应的版本,https://pytorch.org/get-started/previous-versions/ 历届版本在这个链接,注意对应的cuda,也可以看你下载pytorch大小判断是否是gpu版本,我下的gpu有1.5G,cpu版本才200多M。