本地vscode安装GPU版本PyTorch

操作系统 windows, IDE环境vscode,本地GPU
可以新建一个jupyter文件,运行一些测试代码

  • 确保装好显卡驱动

在底下调出终端窗口,默认是power shell,我喜欢用cmd窗口
激活自己的虚拟环境,输入命令 nvidia-smi
本地vscode安装GPU版本PyTorch_第1张图片

  • 确保自己安装好了CUDA
!nvcc --version

本地vscode安装GPU版本PyTorch_第2张图片
安装好之后输入上述代码会显示CUDA版本

  • 安装pytorch

本地vscode安装GPU版本PyTorch_第3张图片
去官网复制他的代码运行。我用conda的代码会显示一个SSL的错误,我用pip安装会给我装一个cpu版本的pytorch,

import torch
print(torch.__version__)

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

  • 检验一下

本地vscode安装GPU版本PyTorch_第4张图片
本地vscode安装GPU版本PyTorch_第5张图片

你可能感兴趣的:(vscode,pytorch,ide)