1.查看cuda版本,我是11.6.134
原来是下载成了cpu版本的torch
卸载
pip uninstall torch
选择合适的版本
cu116表示cuda11.6,cp37表示python3.7,win_amd64表示windows64位系统
download.pytorch.org/whl/torch_stable.html
可以了……
1.查看cuda版本,我是11.6.134
原来是下载成了cpu版本的torch
卸载
pip uninstall torch
选择合适的版本
cu116表示cuda11.6,cp37表示python3.7,win_amd64表示windows64位系统
download.pytorch.org/whl/torch_stable.html
可以了……