本文有用的前提：你是用Anaconda装的python开发环境，用pytorch构建模型。
如何用Anaconda装的python开发环境，请参考另一篇博客，链接如下
Anaconda + Pycharm，利用Anaconda安装python并配置虚拟环境，包括sklearn、pytorch的安装_learninger_lt7的博客-CSDN博客
一、Win + R， cmd —> 输入
nvidia-smi找到
这里的CUDA Version指的是你的电脑最高支持的CUDA版本，它是向下兼容的，因此，低于12.0版本的CUDA都可安装。
二、进入pytorch官网，PyTorch
往下滑，找到这个，按我这么选
建议选择CUDA11.7版本，我试过11.8，莫名的报错。
扫描二维码关注公众号，回复：
15598374 查看本文章
复制灰色部分的代码。
三、打开开始菜单，找到Anaconda Powershell Prompt，点击进入
输入：conda activate myEnvs_3_7_16 # (红色部分应根据你自己的虚拟环境名填写)
成功激活虚拟环境后，接着在红线处粘贴刚刚复制的那段代码，然后回车等待安装即可。
如果提示pip版本问题，就在当前界面（虚拟环境已激活的情况下）升级pip，升级pip的代码如下：
python -m pip install --upgrade pip