pytorch安装(windows)

B站视频讲解:https://www.bilibili.com/video/BV1VC4y1o7c4/?spm_id_from=333.999.0.0&vd_source=b183d959efa43298bb324ccf00fb30a5

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的版本也对应。

torch 和 torchvision 的版本对应关系:https://pytorch.org/get-started/previous-versions/

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

3. torch验证

1)验证GPU是否有效

import torch 
print(torch.cuda.is_available()) 

2) 查询PyTorch版本

import torch 
print(torch.__version__) 

4. 报错处理

ERROR: torchvision-0.1,+cu111-cp38-cp38win_amd64 ,whl is not a supported wheel on this platform.

请添加图片描述
输入pip debug --verbose命令查看平台支持的版本,结果如下图:

在这里插入图片描述
按照自己的版本修改命名即可。

我修改后的文件名如下:torch-1.12.1+cu116-cp39-abi3-win_amd64.whl,其实对照上图中的任何一个的名称都可以,然后再用pip install torch-1.12.1+cu116-cp39-abi3-win_amd64.whl 安装,成功。

来源查看博客 https://blog.csdn.net/qq_44832009/article/details/129351554解决

  • 5
    点赞
  • 42
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Python图像识别

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值