本地vscode安装GPU版本PyTorch

寂静的以2023-10-12 18:54

操作系统 windows, IDE环境vscode，本地GPU

可以新建一个jupyter文件，运行一些测试代码

在底下调出终端窗口，默认是power shell，我喜欢用cmd窗口

激活自己的虚拟环境，输入命令 nvidia-smi

python 复制代码

!nvcc --version

安装好之后输入上述代码会显示CUDA版本

去官网复制他的代码运行。我用conda的代码会显示一个SSL的错误，我用pip安装会给我装一个cpu版本的pytorch，

python 复制代码

import torch
print(torch.__version__)

他输出结果torch2.1.0+cpu，就是说我装的实际上是cpu版本的。所以不要用pip,用conda 。我重装了一遍anaconda，（原本建的虚拟环境不会丢，不用担心），终于把pytorch装上去了。