本地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装上去了。

  • 检验一下


相关推荐
rainFFrain6 小时前
单例模式与线程安全
linux·运维·服务器·vscode·单例模式
誉鏐7 小时前
PyTorch复现线性模型
人工智能·pytorch·python
牙牙要健康9 小时前
【目标检测】【深度学习】【Pytorch版本】YOLOV3模型算法详解
pytorch·深度学习·目标检测
超级小的大杯柠檬水9 小时前
修改Anaconda中Jupyter Notebook默认工作路径的详细图文教程(Win 11)
ide·python·jupyter
色空大师11 小时前
【idea】实用插件
java·ide·intellij-idea
誉鏐11 小时前
PyTorch复现逻辑回归
人工智能·pytorch·逻辑回归
suanday_sunny11 小时前
VSCode运行,各类操作缓慢,如何清理
ide·vscode·编辑器
信计小白12 小时前
vscode报Module containing this breakpoint has not yet loaded
ide·vscode·编辑器
意.远12 小时前
在PyTorch中使用GPU加速:从基础操作到模型部署
人工智能·pytorch·python·深度学习
MonkeyKing_sunyuhua21 小时前
Visual Studio Code 进行汉化
ide·vscode·编辑器