本文有用的前提：你是用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，莫名的报错。 

 复制灰色部分的代码。

三、打开开始菜单，找到Anaconda Powershell Prompt，点击进入

 输入：conda activate myEnvs_3_7_16  # (红色部分应根据你自己的虚拟环境名填写)

成功激活虚拟环境后，接着在红线处粘贴刚刚复制的那段代码，然后回车等待安装即可。

如果提示pip版本问题，就在当前界面（虚拟环境已激活的情况下）升级pip，升级pip的代码如下：

python -m pip install --upgrade pip