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__) 
相关推荐
DolphinScheduler社区几秒前
白鲸开源与亚马逊云科技携手推动AI-Ready数据架构创新
人工智能·科技·开源·aws·白鲸开源·whalestudio
欣然~29 分钟前
借助 OpenCV 和 PyTorch 库,利用卷积神经网络提取图像边缘特征
人工智能·计算机视觉
岫珩29 分钟前
“由于启动计算机时出现了页面文件配置问题,Windows在你的计算机上创建了一个临时页面文件。。。”的问题解决
windows
谦行38 分钟前
工欲善其事,必先利其器—— PyTorch 深度学习基础操作
pytorch·深度学习·ai编程
李菠菜1 小时前
解决Windows系统下Git克隆时报错“unable to checkout working tree”的方法详解
windows·git
白熊1881 小时前
【计算机视觉】CV实战项目 - 基于YOLOv5的人脸检测与关键点定位系统深度解析
人工智能·yolo·计算机视觉
nenchoumi31191 小时前
VLA 论文精读(十六)FP3: A 3D Foundation Policy for Robotic Manipulation
论文阅读·人工智能·笔记·学习·vln
后端小肥肠1 小时前
文案号搞钱潜规则:日入四位数的Coze工作流我跑通了
人工智能·coze
LCHub低代码社区1 小时前
钧瓷产业原始创新的许昌共识:技术破壁·产业再造·生态重构(一)
大数据·人工智能·维格云·ai智能体·ai自动化·大禹智库·钧瓷码
-曾牛1 小时前
Spring AI 快速入门:从环境搭建到核心组件集成
java·人工智能·spring·ai·大模型·spring ai·开发环境搭建