深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录

这篇具有很好参考价值的文章主要介绍了深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

今天不小心将conda环境中的一个pytorch环境中的torch包给搞混了,将其更新了一下,发生了一些问题:

当时运行了一下这个代码: pip install torchvision --upgrade

导致了环境中包的混乱:

深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

只能说欲哭无泪,当时这个 pytorch环境中我是安装的CUDA11.8的版本应该,后来安装了cpu版本的将原先的给覆盖了,导致运行代码时遇到如下报错:

ValueError: Could not find the operator torchvision::nms. Please make sure you have already registered the operator and (if registered from C++) loaded it via torch.ops.load_library.

遇到“Could not find the operator torchvision::nms”这类错误,通常是因为torchtorchvision版本之间的不兼容问题。您提供的信息显示,您的torch版本是2.1.2+cu118。这个问题可能因为torchvision版本不匹配或者安装有问题导致。

这次无意的更新导致了版本的冲突问题!!!

只能重新安装一遍了,先查看显卡配置:

nvidia-smi

深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

 我的CUDA Version:12.1

前往Pytorch官网进行安装:Start Locally | PyTorch

深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

复制这个安装命令:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

在pycharm的终端处可以进行安装,输入如下指令:

conda activate pytorch

 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

安装时间比较长,耐心等待。。。。。。。

安装结束后,查看torch与torchvision的cuda版本:

pip show torch torchvision

深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

验证安装

安装完成后,您可以通过运行以下Python代码来验证torchvision是否正确安装,并且是否能够与torch一起使用CUDA:

import torch
import torchvision

print("Torch version:", torch.__version__)
print("Torchvision version:", torchvision.__version__)
print("CUDA available:", torch.cuda.is_available())
if torch.cuda.is_available():
    print("CUDA version:", torch.version.cuda)

我的测试结果:
深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

说明安装成功!!!

 最后测试一下代码运行:
深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录,深度学习,Python,深度学习,pytorch,人工智能,python

没有报错,又可以继续学习Pytorch了!!! 文章来源地址https://www.toymoban.com/news/detail-828676.html

到了这里,关于深度学习:Pytorch安装的torch与torchvision的cuda版本冲突问题与解决历程记录的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • PYTORCH 依赖 cuda 11.8 , cuda 12.1 cpu 里 torch torchvision torchaudio对应关系

    Pytorch是Facebook人工智能研究院基于Torch开发的一个开源的Python机器学习库,它提供了两项强大的功能: 1、强大的GPU加速的张量计算(NumPy)。 2、自动求导系统的深度神经网络。 cuda 11.8 cuda 12.1 cpu  Previous PyTorch Versions | PyTorch

    2024年02月01日
    浏览(54)
  • 2022-2023最新 pytorch安装方法 GPU版本 python3.9 torch-1.13.0+cu116-cp39 torchvision-0.14.0 亲自安装可用!(其他版本也通用)

    如图,这样是不能安装gpu版本的。 这里针对python3.9版本,在此网站https://github.com/pytorch/vision查找读者python版本对应的torch和torchvision。 然后在这个网址https://download.pytorch.org/whl/torch/ 和 https://download.pytorch.org/whl/torchvision/里下载,我这里是对应了这两个。千万要对应,不然会报

    2024年02月06日
    浏览(51)
  • 【安装GPU版本pytorch,torch.cuda.is_available()仍然返回False问题】

    @TOC 检查cuda是否安装,CUDA环境变量是否正确设置,比如linux需要设置在PATH,window下环境变量编辑看看,是否有CUDA 首先查看你环境里的pytorch是否是cuda版本,我这里是conda安装的,使用conda list查看是这样: 圈出来的torch带cuda,那安装的肯定时cuda版本,但是torch.cuda.is_available

    2024年02月10日
    浏览(46)
  • PyTorch中torch、torchtext、torchvision、torchaudio与Python版本兼容性

    torch与torchtext,Python对应关系,来源:https://pypi.org/project/torchtext/ 截止发文,最新版本:torch 2.0.0,torchtext 0.15.1 安装方法: 或 torch与torchvision,Python对应关系,来源:https://github.com/pytorch/vision 截止发文,最新版本:torch 2.0.0,torchvision 0.15.1 安装方法: 或 torch与torchaudio,Pyt

    2024年02月04日
    浏览(73)
  • 深度学习|如何确定 CUDA+PyTorch 版本

    对于深度学习初学者来说,配置深度学习的环境可能是一大难题,因此本文主要讲解CUDA; cuDNN; Pytorch 三者是什么,以及他们之间的依赖关系。 CUDA(Compute Unified Device Architecture)是由NVIDIA开发的用于并行计算的平台和编程模型。CUDA旨在利用NVIDIA GPU(图形处理单元)的强大计算

    2024年02月07日
    浏览(45)
  • 集显安装pytorch教程,cuda和cudann环境配置torch-GPU版本看这一篇就够了

    集显看最下面 首先GPU安装教程 1.安装Anaconda (这个是可以安装环境和前置的软件),这个网址是清华源下载,可以去官网下载 Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror 2.安装CUDA (这个是显卡驱动,需要安装好以用显卡计算,缩短运行时间)(c

    2024年02月02日
    浏览(72)
  • Python安装torch(含torch和torchvision对应版本)

    torch 、torchvision和python的对应版本如下: torch  torchvision  python master / nightly master / nightly =3.6 1.7.1  0.8.2 =3.6 1.7.0  0.8.1 =3.6 1.7.0 0.8.0 =3.6 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, =3.7 1.2.0 0.4.0 ==2.7, =3.5, =3.7 1.1.0  

    2024年01月16日
    浏览(56)
  • PyTorch框架中torch、torchvision、torchaudio与python之间的版本对应关系(9月最新版)

    随着python语言和pytorch框架的更新,torchtorchvisiontorchaudio与python之间的版本对应关系也在不断地更新。 最新版本 torch与torchvision 对应关系如下: 稍旧版本 torch与torchvision 对应关系如下: 最新版本 torch与torchaudio 对应关系如下:

    2024年02月21日
    浏览(47)
  • Windows 下载与安装CUDA和Pytorch【安装教程、深度学习】

    参考链接:Windows 下安装 CUDA 和 Pytorch 跑深度学习 - 动手学深度学习v2_哔哩哔哩_bilibili 请确保你是NVIDIA的显卡( 不能是AMD 、 集成显卡 ) 打开developer.nvidia.com/cuda-downloads,打开有点慢 选择Windows 本地安装 接下来就会出现对应的安装包 安装或许会有点慢,取决你网速 下载完成

    2024年02月07日
    浏览(53)
  • 安装对应版本pytorch和torchvision

    遇见报错: ERROR: Could not find a version that satisfies the requirement torch (from versions: none) ERROR: No matching distribution found for torch 解决方法: 1、网站找到对应torch和torchvision版本,cp对应python版本,cu对应的是gpu版本的torch:https://download.pytorch.org/whl/torch_stable.html 2、下载对应版本 3、在下

    2024年02月10日
    浏览(40)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包