在官网下载的gpu版本torch总是cpu版
因为这不是真的gpu版!!!
搞了好久简直要疯掉555555
解决方案
通过本地whl文件的方式安装。
来到该网站:https://download.pytorch.org/whl/torch
可以看到各个版本的torch,xuanze符合自己版本的就行
torch-1.13.0+cu116-cp37-cp37m-win_amd64.whl,意为torch1.13.0、cuda11.6、python3.7、win 64位系统的版本。
下载到本地后安装就可以
pip3 install D:\torch-1.13.0+cu116-cp37-cp37m-win_amd64.whl