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__) 
相关推荐
我爱一条柴ya14 分钟前
【AI大模型】神经网络反向传播:核心原理与完整实现
人工智能·深度学习·神经网络·ai·ai编程
万米商云18 分钟前
企业物资集采平台解决方案:跨地域、多仓库、百部门——大型企业如何用一套系统管好百万级物资?
大数据·运维·人工智能
新加坡内哥谈技术22 分钟前
Google AI 刚刚开源 MCP 数据库工具箱,让 AI 代理安全高效地查询数据库
人工智能
慕婉030723 分钟前
深度学习概述
人工智能·深度学习
大模型真好玩24 分钟前
准确率飙升!GraphRAG如何利用知识图谱提升RAG答案质量(额外篇)——大规模文本数据下GraphRAG实战
人工智能·python·mcp
198925 分钟前
【零基础学AI】第30讲:生成对抗网络(GAN)实战 - 手写数字生成
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·近邻算法
6confim25 分钟前
AI原生软件工程师
人工智能·ai编程·cursor
阿里云大数据AI技术26 分钟前
Flink Forward Asia 2025 主旨演讲精彩回顾
大数据·人工智能·flink
i小溪27 分钟前
在使用 Docker 时,如果容器挂载的数据目录(如 `/var/moments`)位于数据盘,只要服务没有读写,数据盘是否就不会被唤醒?
人工智能·docker
程序员NEO29 分钟前
Spring AI 对话记忆大揭秘:服务器重启,聊天记录不再丢失!
人工智能·后端