惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

V
Visual Studio Blog
Microsoft Azure Blog
Microsoft Azure Blog
WordPress大学
WordPress大学
小众软件
小众软件
Last Week in AI
Last Week in AI
月光博客
月光博客
博客园 - 聂微东
Recent Announcements
Recent Announcements
A
About on SuperTechFans
博客园 - 三生石上(FineUI控件)
V
V2EX
阮一峰的网络日志
阮一峰的网络日志
博客园 - Franky
云风的 BLOG
云风的 BLOG
量子位
N
Netflix TechBlog - Medium
Hugging Face - Blog
Hugging Face - Blog
H
Hackread – Cybersecurity News, Data Breaches, AI and More
J
Java Code Geeks
博客园 - 司徒正美
S
SegmentFault 最新的问题
有赞技术团队
有赞技术团队
Google DeepMind News
Google DeepMind News
宝玉的分享
宝玉的分享

博客园 - ubirdy

在Cursor中使用Anthropic的Skills 切换至windows自带中文输入法后卡住的bug解决 用ikuai软路由提供内网NTP服务 在欧拉系统(openEuler)上安装uWSGI 在欧拉系统(openEuler)上安装Nginx 海康工业相机SDK错误码常见场景解析 ubuntu22.04 server安装nvidia cuda 12.08提示Nouveau 驱动已经在用 ubuntu22.04更换镜像源 ubuntu22.04安装CUDA ICMP时间戳漏洞CVE-1999-0524的处理 解决supervisor在香橙派AIpro上启动问题 用ffplay在香橙派AIPro上播放PCM文件 香橙派OrangePi_AiPro超频CPU与NPU 昇腾310B4在转换onnx格式至om格式时提示BrokenPipeError: [Errno 32] Broken pipe Python打包工具 Pyinstaller使用教程 FFplay命令参数 yoloV7训练结果分析 Win11禁用右键菜单折叠改用经典右键菜单 Altas产品查询CANN软件包版本的方法
在windows中安装pytorch的GPU版本及torchvision
ubirdy · 2025-03-06 · via 博客园 - ubirdy

一. 查看自己电脑的CUDA版本

在powershell中输入:nvidia-smi

 当前CUDA版本为12.6,表示支持最高版本的CUDA是12.6,可以向下兼容低版本的CUDA

二. 下载安装CUDA

NVIDIA官网: https://developer.nvidia.com/cuda-toolkit-archive

用上面的网址下载适配的版本。我用的是windows11,用11.6这个版本。

 下载好之后默认安装即可

三. 查看环境变量

安装完成后查看系统变量:

在“系统属性”--“环境变量”中查看

 在powershell输入nvcc --version验证安装

四. 安装GPU版本的Pytorch和torchvision

pytorch官网的老版本安装链接:https://pytorch.org/get-started/previous-versions/

 我用pycharm做IDE,是用pip安装,复制对应这行命令安装。其它的开发环境对应复制官网的命令。

 也可以用离线下载的方式下载安装,Pytorch下载链接:https://download.pytorch.org/whl/torch_stable.html.

在powershell中输入pip list验证

 验证开发环境

 若 torch.cuda.is_available() 结果为True,则表示GPU版本安装完成