pytorch，torch，torchvision的gpu版本安装避坑

最新推荐文章于 2024-08-08 23:29:37 发布
最新推荐文章于 2024-08-08 23:29:37 发布
阅读量1.6w 收藏 61
点赞数 22
分类专栏： torch 文章标签： pytorch gpu pip
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_46415031/article/details/114809650
版权
本文详细介绍了如何在Python 3.6环境中安装与GPU兼容的PyTorch、Torch和Torchvision。推荐离线安装，通过下载对应版本的whl文件，使用pip在conda环境下安装。安装前需确保已安装正确版本的显卡驱动、CUDA和CuDNN。文章提供了国内镜像源以加速在线安装，但未指定GPU版本可能导致安装CPU版本。
摘要由CSDN通过智能技术生成

pytorch，torch，torchvision的gpu版本安装避坑

在线or离线安装

最近使用pytorch跑深度学习模型，需要用到与gpu相对应的包，写此博文，提醒后来者避避坑

pytorch

本次使用python3.6（conda环境），一般conda环境可以创建多个Python版本，互不干扰，比单一Python要好用，在pycharm上添加pytorch库。
在这里插入图片描述

点击左下方 + ，进去页面输入pytorch，安装对应的版本

torch和torchvision

torch，torchvision库在pycharm中没搜到，在conda环境下激活python36，使用pip命令安装
在使用计算机的狗之前，要安装好显卡驱动，cuda和处cudnn，请至
https://blog.csdn.net/cashmood/article/details/105081586

因为要是用gpu跑代码，所以和torch，torchvision相关的包都必须是和gpu相对应的，

（1）离线安装（强烈推荐）

下载对应的torch和torchvision ：http://download.pytorch.org/whl/torch_stable.html

或者：（第三方扩展包&#x

最低0.47元/天 解锁文章
weixin_46415031
关注
专栏目录
Pytorch GPU版详细安装过程
南柯一梦的江湖
06-15 1408
可以在https://pytorch.org/get-started/previous-versions/ 查询对应版本的torchaudio。可以在https://github.com/pytorch/vision#installation，查询对应版本的torchvision，注意：只有带NVIDIA的英伟达显卡的电脑才能安装GPU版本的Pytorch，否则只能安装CPU版本的Pytorch.然后在https://download.pytorch.org/whl/torch_stable.html。
【深度学习】torch和torchvision的对应安装版本
m0_62920653的博客
07-09 3064
PyTorch 中的 `torch` 和 `torchvision` 版本需要相互匹配。torch、torchvision对应版本。
gpu版本torch安装踩坑总结
最新发布
江湖偌大的博客
08-08 1410
0.我这次最开始是直接在 anaconda prompt 中下载 torch ，用的下面的 pip 命令，下载的实在是太慢太慢了，每次下载torch 都提示要将近三个小时才能下载下来，但是一般下了一小时就报错了。这个方法我之前就试过，但是之前没成功，报错在这里，不知道这次为啥成功了，而且提示很奇怪，好像和其他博主的都不一样。由于我的服务器那边网络极差，直接用官网指令下载不下来，只能用 .whl 文件下载，最终成功了。3.conda版本太低了4.10.3，更新到了22.9.0，继续下载，还是报错。
torch1.8+torchvision-0.9.zip
08-21
用于树莓派官方32位arrm7l系统配置pytorch环境，可用于YOLOV5环境的搭建
树莓派4B部署yolov5：安装torch：torch1.8.0+torchvision0.9.0
syh的技术记录
08-12 4707
前两天一直在树莓派4B上部署yolov5，但就是一直安装不好要求版本的pytorch，难受坏了。 官方要求版本： 这里分享编译好的torch1.8.0+torchvision0.9.0的.whl文件，下载好直接pip3 install XXX.whl即可。 链接：https://pan.baidu.com/s/1J-PPpp_y2U-Qon3IEi7nNA 提取码：qwer 资源来自：https://github.com/sungjuGit/PyTorch-and-Vision-for-Raspberry
jetsonnano安装torch1.8.0与torchvision0.9.0（个人血泪史）
cyh20182808的博客
05-23 1947
然后重启了一下，重新来一遍咱们的操作，从进torchvision这个文件夹开始，然后就，就安装成功了，我吃完饭回来看到成了，高兴的想给它磕两个，安了好多天，各种各样的torchvision就是怎么都安不上，好了不叭叭了。注意了，这里我输入这条命令才好使（加了sudo的不好使，说我非法指令，怎么改都不行）里面有一个whl文件的torch，与一个torchvision的文件夹。然后呢，开始编译了，那么最后可能会报错，例如我就报了这个。的命令端，进入这个文件夹中，命令就是。我的板板是4.6，cuda10.2。
【EdgeBox_tx1_tx2_E100】 PyTorch v1.8.0 torchvision v0.9.0 环境部署
cau_weiyuhu的博客
01-04 1967
介绍PyTorch 环境 在 EHub_tx1_tx2_E100载板，TX1核心模块环境（Ubuntu18.04）下如何实现部署和测试，准备安装的环境是（PyTorch v1.8.0 + torchvision v0.9.0）。
Jeston nano pytorch1.8.0 torchvision0.9.0
weixin_46735493的博客
11-15 974
Jeston nano
Pytorch、torch、torchvision的cpu和gpu版本的安装
sinat_37281674的博客
09-21 3387
Pytorch的cpu和gpu版本的安装CPU版本的安装GPU版本的安装 CPU版本的安装 通过命令行自动下载安装的方法是不可行的，因为下载总是找不到资源或者下载失败，所以只能通过命令行的方式进行安装。 这个链接是各种版本torch的.whl方式，可以手动进行下载，然后本地安装 https://download.pytorch.org/whl/torch_stable.html GPU版本的安装 通过命令行自动下载安装的方法是不可行的，因为下载总是找不到资源或者下载失败，所以只能通过命令行的方式进行安装。
torchvision0.8.2 gpu版 win64 python3.7 cuda10.1
04-20
为方便国内Windows用户下载安装gpu版的torch1.7.1和torchvision0.8.2，现上传从官网下载的.whl安装包资源至我们广博的C站~（这部分是torchvision）
安装GPU版本Pytorch安装GPU版本Pytorch
10-28
在本文中，我们将详细讨论如何在计算机上安装GPU版本的PyTorch，这是一个深度学习框架，广泛用于构建和训练神经网络。PyTorch的GPU版本能够利用图形处理单元（GPU）的强大计算能力来加速深度学习模型的训练过程。 ...
在 NVIDIA Jetson上安装GPU版本的pytorch和torchvision
NotName_9758的博客
02-01 910
这些pytorch包仅支持python3.6，使用conda直接创建python3.6时会报错，修改命令为。参考链接【3】处下载torch安装包。来正确创建python3.6环境。
torch对应的torchvision版本
热门推荐
学无止境、积少成多、厚积薄发
10-12 1万+
安装torch和torchvision时，两者之间存在依赖关系，版本需要对应起来。 torch torchvision python 1.6.0 0.7.0 >=3.6 1.5.1 0.6.1 >=3.5 1.5.0 0.6.0 >=3.5 1.4.0 0.5.0 ==2.7, >=3.5, <=3.8 1.3.1 0.4.2 ==2.7, >=3.5, <=3.7 1.3.0 0.4.1 ==2.7, >=3.5, &lt
pip install torch及对应torchvision、torchaudio
weixin_43994934的博客
07-12 3340
在终端Terminal 中输入以上结果后，就自动帮你安装了。
Jetson Xavier安装pytorch 1.8.0+torchvision0.9.0
耶律大石的博客
01-10 2279
语义分割学习笔记2——Jetson Xavier安装pytorch一、whl文件安装pytorch二、安装torchvision三、测试是否安装成功 我的环境说明： 名称 版本 硬件： Nvidia Jetson Xavier AGX CUDA版本： 10.2 Jetpack版本： JP 4.5 python版本： python2.7 python3.6 deepstream-app version 5.0.0 TensorRT Version: 7.1 cuDNN
torchvision.transforms GPU加速，提升预测阶段数据预处理速度
胖胖大海的博客
01-20 7087
参考：Tensor transforms and JIT — Torchvision 0.11.0 documentation Torchvision v0.8.0之前版本： Torchvision v0.8.0之前版本的transforms主要分为两类： 1、一类处理的输入数据类型为Tensor 2、另一类处理的数据类型为PILImage 所以torchvision里面提供了两个转换函数ToTensor和ToPILImage...
记录win10系统pip本地离线安装torch1.8.0+cu111（python37)
s13679203258的博客
11-14 2434
pytorch安装
pytorch和torchvision cuda版本
09-04
PyTorch和Torchvision都支持CUDA加速，可以在GPU上运行。PyTorch是一个用于深度学习的开源机器学习框架，而Torchvision是PyTorch的一个计算机视觉库，提供了一系列图像和视频处理工具。 要使用CUDA加速，首先需要确保你的系统上安装了合适的NVIDIA显卡驱动和CUDA工具包。然后，安装正确版本的PyTorch和Torchvision。 你可以通过下面的命令安装最新版本的PyTorch： ``` pip install torch torchvision ``` 这将会安装最新的CPU版本。如果你想要安装支持CUDA加速的版本，需要根据你的CUDA版本选择合适的PyTorch版本。例如，如果你的CUDA版本是10.2，可以使用以下命令安装PyTorch： ``` pip install torch==1.7.1+cu102 torchvision==0.8.2+cu102 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html ``` 这将会安装PyTorch 1.7.1、Torchvision 0.8.2 和相应的CUDA 10.2版本。 请注意，要正确配置CUDA并启用GPU加速，你需要确保符合系统要求，并按照PyTorch的文档进行设置。此外，你还可以在PyTorch官方网站上找到有关CUDA支持和安装说明的更多信息。
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值