用官网指令安装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