下载whl文件
提示:下载torch的稳定版本网址[https://download.pytorch.org/whl/torch_stable.html](https://download.pytorch.org/whl/torch_stable.html)
首先,查看主机显卡和安装的cuda版本。
我的为12.1版本。
若没有,自行安装cuda,cudann。
在conda中创建虚拟环境 torch,我已创建,如下图。
然后,根据配置好的环境下载一个对应版本的pytorch
没显卡就下载cpu的,安装显卡,就下载cu的。我下载cu92,打开网址,Ctrl+F,搜索cu121,回车。