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

推荐订阅源

博客园 - Franky
D
Docker
Jina AI
Jina AI
The GitHub Blog
The GitHub Blog
博客园 - 聂微东
B
Blog RSS Feed
大猫的无限游戏
大猫的无限游戏
M
MIT News - Artificial intelligence
Vercel News
Vercel News
Microsoft Security Blog
Microsoft Security Blog
博客园 - 叶小钗
爱范儿
爱范儿
D
DataBreaches.Net
Hugging Face - Blog
Hugging Face - Blog
IT之家
IT之家
Recent Announcements
Recent Announcements
U
Unit 42
腾讯CDC
奇客Solidot–传递最新科技情报
奇客Solidot–传递最新科技情报
宝玉的分享
宝玉的分享
量子位
Stack Overflow Blog
Stack Overflow Blog
Cyber Security Advisories - MS-ISAC
Cyber Security Advisories - MS-ISAC
Microsoft Azure Blog
Microsoft Azure Blog

博客园 - 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版本安装完成