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__) 
相关推荐
wwlsm_zql2 小时前
「赤兔」Chitu 框架深度解读(十四):核心算子优化
人工智能·1024程序员节
AKAMAI4 小时前
Fermyon推出全球最快边缘计算平台:WebAssembly先驱携手Akamai云驱动无服务器技术新浪潮
人工智能·云计算·边缘计算
云雾J视界5 小时前
TMS320C6000 VLIW架构并行编程实战:加速AI边缘计算推理性能
人工智能·架构·边缘计算·dsp·vliw·tms320c6000
想ai抽5 小时前
基于AI Agent的数据资产自动化治理实验
人工智能·langchain·embedding
小马过河R6 小时前
AIGC视频生成之Deepseek、百度妙笔组合实战小案例
人工智能·深度学习·计算机视觉·百度·aigc
june-Dai Yi6 小时前
免费的大语言模型API接口
人工智能·语言模型·自然语言处理·chatgpt·api接口
王哈哈^_^7 小时前
【数据集】【YOLO】【目标检测】农作物病害数据集 11498 张,病害检测,YOLOv8农作物病虫害识别系统实战训推教程。
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·1024程序员节
数据库安全7 小时前
牛品推荐|分类分级效能飞跃:美创智能数据安全分类分级平台
大数据·人工智能·分类
却道天凉_好个秋7 小时前
卷积神经网络CNN(六):卷积、归一化与ReLU总结
人工智能·神经网络·cnn
澄澈青空~7 小时前
blender拓扑建模教程
人工智能·blender