pytorch安装(windows)

1. torch 简介

torch总共分为两个版本,GPU版和CPU版,CPU版安装非常简单,直接pip install 安装即可,在此详细介绍 GPU 版的安装方式。

GPU 版安装需要注意以下几个地方,需要进行匹配才能安装上,否则即便安装上也运行不了

(1)CUDA的版本

(2)python的版本

(3)操作系统,本博客以 windows 为例

2. torch安装

具体的安装步骤如下:

1) pytorch 官网下载https://download.pytorch.org/whl/torch_stable.html

打开上面链接,界面如下,以下包含torch、torchvison、torchaudio 等相关包的各个版本,其中每个数字代表的含义入下图:

找到自己想要下载的版本,点击即可下载。
注意cuda的版本、python的版本、以及windows/linux 这些必须对应好,否则即便安装上,也不能调用cuda。

我们需要下载 torchtorchvision 两个关键包,torchaudio是通音频相关的包,不搞音频的话,可以不用安装。

torch 和 torchvision 的版本一定要对应, python 、cuda的版本也对应。

2)下载完成后,得到的是 whl文件, 然后打开文件所在文件夹,在路径下输入cmd,打开 命令提示符, 输入 pip install xxx.whl 进行安装,先安装 torch包、再安装 torchvision包。

3. torch验证

1)验证GPU是否有效

python 复制代码
import torch 
print(torch.cuda.is_available()) 

2) 查询PyTorch版本

python 复制代码
import torch 
print(torch.__version__) 
相关推荐
新智元15 分钟前
复旦张军平:人类是硅基生命过渡体?
人工智能·openai
新智元21 分钟前
谷歌Gemini 3.0「全家桶」年度压轴,前端不再需要人类!下周王者降临
人工智能·openai
机器之心35 分钟前
ICLR 2026惊现SAM 3,分割一切的下一步:让模型理解「概念」
人工智能·openai
清风吹过38 分钟前
LSTM新架构论文分享6:LSTM+Transformer融合
论文阅读·人工智能·深度学习·神经网络·lstm·transformer
Hoking44 分钟前
LangChain4j集成SpringBoot接入百炼大模型(Qwen)
java·人工智能·spring boot·llm
量子位1 小时前
马斯克从英伟达挖人做AI游戏!第一步:研发世界模型
人工智能·ai编程
程序员大雄学编程1 小时前
「机器学习笔记11」深入浅出:解密基于实例的学习(KNN算法核心原理与实践)
人工智能·笔记·机器学习
WTCLLB1 小时前
Win10,在ESP分区添加PE系统,隐藏VTOYEFI分区
windows·ventoy·bcd·wepe
AI拉呱_1 小时前
第145期《2025年AI现状报告》解读(三):安全篇
人工智能·安全
益企研究院1 小时前
AICC2025 智算中心与算力服务论坛顺利举办
人工智能