本地vscode安装GPU版本PyTorch

操作系统 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装上去了。

• 检验一下