本地vscode安装GPU版本PyTorch

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

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

  • 确保装好显卡驱动

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

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

  • 确保自己安装好了CUDA
python 复制代码
!nvcc --version

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

  • 安装pytorch

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

python 复制代码
import torch
print(torch.__version__)

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

  • 检验一下


相关推荐
l12345sy10 小时前
Day24_【深度学习—广播机制】
人工智能·pytorch·深度学习·广播机制
CAE虚拟与现实1 天前
VSCode中的下载VSIX是指什么?
ide·vscode·编辑器
路边闲人21 天前
vscode启用GEMINI CODE ASSIST插件
ide·vscode·gemini
小蕾Java1 天前
Java 开发工具,最新2025 IDEA使用(附详细教程)
java·ide·intellij-idea
CAE虚拟与现实1 天前
VSCode官方汉化包
ide·vscode·编辑器·vscode汉化
CAE虚拟与现实1 天前
VSCode创建Python项目和运行py文件
ide·vscode·编辑器
资讯第一线1 天前
《RAD Studio 13.0》 [DELPHI 13.0] [官方原版IOS] 下载
ide
蒋星熠1 天前
如何在Anaconda中配置你的CUDA & Pytorch & cuNN环境(2025最新教程)
开发语言·人工智能·pytorch·python·深度学习·机器学习·ai
过-眼-云-烟1 天前
新版Android Studio能打包但无法run ‘app‘,编译通过后手机中没有安装,顶部一直转圈
android·ide·android studio
Stardep1 天前
ssh远程连接服务器到vscode上“连接失败”
服务器·vscode·ssh