操作系统 windows, IDE环境vscode，本地GPU
可以新建一个jupyter文件，运行一些测试代码
- 确保装好显卡驱动
在底下调出终端窗口，默认是power shell，我喜欢用cmd窗口
激活自己的虚拟环境，输入命令 nvidia-smi
- 确保自己安装好了CUDA
!nvcc --version
安装好之后输入上述代码会显示CUDA版本
- 安装pytorch
去官网复制他的代码运行。我用conda的代码会显示一个SSL的错误，我用pip安装会给我装一个cpu版本的pytorch，
import torch
print(torch.__version__)
他输出结果torch2.1.0+cpu，就是说我装的实际上是cpu版本的。所以不要用pip,用conda 。我重装了一遍anaconda，（原本建的虚拟环境不会丢，不用担心），终于把pytorch装上去了。
- 检验一下