本文介绍了如何在官方指令下安装CPU版本的PyTorch，以及如何切换到GPU版本（如cu113）以适应cuda11.3。重点提到了版本选择和兼容性，并提供了检测PyTorch版本的代码。还提到由于paddlepaddle限制，切换到cuda11.2和torch-1.10.2+cu113的解决方案。
用官网指令安装torch是cpu版本的。 不知道安装的是cpu还是gpu版本对可以用torch.__version__查看,如果是xx+cpu就是cpu版本。
如果要安装GPU版本，去torch_stable里面找对应版本. 这里我用的是cu113/torch-1.10.2%2Bcu113-cp36-cp36m-win_amd64.whl理解一下参数：cu113表示需要cuda11.3，torch版本是1.10,2，cp36指需要的是python3.6，win系统.
另外如果之前有装过cuda的其他版本,再次安装不会覆盖.所以装了新cuda测试nvcc -V 还是老版本。只需要删除老版本的cuda的环境变量就可以.
检测pytorch是否可用的代码:
import torch
print(torch.__version__)
另1: 2022.4.29 由于要使用paddlepaddle,目前其只支持到11.2,故更换为11.2 ,torch1.10.2+cu113可兼容cuda11.2
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
