首页 > 其他分享 >torch版本真的很重要!!!

torch版本真的很重要!!!

时间:2023-12-03 15:13:42浏览次数:34  
标签:模型 torch 参数 cuda 版本 真的 师兄

事情的经过就是,跑深度学习代码的时候,遇到了一系列的错误

参数维度对不上

1.运行时,发现预训练模型得到的参数跟我模型要的对不上,傻逼了,当时没看见github得issues里面就有解答,找了大半天,还尝试去改模型参数。
其实就是因为下载的预训练模型参数的版本不对,应该用旧的版本。

cuda用不了

RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)
先是去调试,发现是代码运行到linear层的时候报错的,然后打印了一下维度,发现一个是二维,一个是三维,但是俺的师兄说不是这个问题,进而作罢。
然后因为上面说是用的cuda,直接在自己创建的环境下使用print(torch.cuda.device_count())是可以打印的,但是放到我跑的代码里头,就报错了,想着应该是这个问题吧。
然后俺就装了cuda和cudann,发现还是没有解决问题!!!
然后我的好心师兄就帮登了我的号去看了一下,给俺重新弄了一个环境,不得不说,我的师兄实在是太好了,类目。
解决方法就是,重新装了一个高一点版本的torch,因为我之前是按照代码里的readme直接装的torch==1.4.0,但是这个实在是太老了,跟俺服务器里的12.0已经匹配不上了,所以装了个新的就好了。

标签:模型,torch,参数,cuda,版本,真的,师兄
From: https://www.cnblogs.com/study2023/p/17873204.html

相关文章

  • 极语言3-8绘制区类、快捷键类、系统信息类、版本信息类、进程信息类、内存信息类——
    中文名字英文名称长度作用解释绘制区类PAINTSTRUCT64包含用来绘制它所拥有的窗口客户区所需要的信息的结构体。绘制区类——成员表设备hdc整数用于绘制的句柄。擦除fErase整数如果为非零值则擦除背景,否则不擦除左边rcPaint.left整数要绘制矩形的左边位置顶部rcPaint.top整数要绘制......
  • 你真的了解HashSet 和HashMap的区别、优缺点、使用场景吗?
     HashSet和HashMap是Java集合框架中的两个常用类,它们都用于存储和管理数据,但在使用方式、功能和性能上有很大的区别。HashSet和HashMap的区别区别一:用途不同HashSet: HashSet是一个基于哈希表的集合,用于存储不重复的元素,它不存储键值对。它实际上是基于HashMap......
  • PyTorch解説
    PyTorch是一种面向Python的开源机器学习库。它是由Facebook的人工智能研究团队基于最初支持多范式脚本语言“Lua”的Torch开发而来。Python是一种广泛用于“利用机器学习进行人工智能开发”、“Web服务和Web应用开发”、“区块链开发”以及“物联网开发”等多个领域的编程语言。......
  • Keras 3.0正式发布:可用于TensorFlow、JAX和PyTorch
    前言 Keras3.0正式发布:可用于TensorFlow、JAX和PyTorch本文转载自机器之心仅用于学术分享,若侵权请联系删除欢迎关注公众号CV技术指南,专注于计算机视觉的技术总结、最新技术跟踪、经典论文解读、CV招聘信息。CV各大方向专栏与各个部署框架最全教程整理【CV技术指南】CV全栈......
  • PieCloudDB Database 再次升级!社区版全新版本发布,免费下载
    10月24日,2023拓数派年度技术论坛在上海圆满落幕。拓数派正式发布大模型数据计算系统「πDataCS」,基于云原生技术重构数据存储和计算,重塑数仓、向量和机器学习等数据计算引擎,实现“一份数据存储,多引擎数据计算”。作为πDataCS首款计算引擎,PieCloudDBDatabase再度升级,推出全新......
  • 微软Windows硬件最新驱动下载地址,无讨论,版本新,都是硬件厂家上传
    下面地址可以下载Windows硬件最新驱动,这里下载的地址是纯粹的驱动,没有其他累赘。这里的驱动是硬件厂家上传的驱动,我查找的是realtek8852AE的驱动,其他任何地方找到的驱动都没有这里的新。realtek8852AE这款螃蟹无线网卡,联想,红米等很多厂家的笔记本都在用,但器旧版驱动容易出......
  • 你真的懂Framework?不懂Framework拿什么去跳槽涨薪啊?
    前言在这个移动应用开发热潮的时代,Android开发者的需求量也在不断增长。而Framework作为Android操作系统的核心组件,对于Android开发者来说具有极其重要的地位。但是,很多开发者却对Framework了解不够深入,甚至存在误解。这种情况不仅影响了开发者的技术水平提升,还对职业发展造成了阻......
  • pytorch 学习记录——计算图
    1.pytorch的计算图是动态更新的(tensorflow是静态计算图),数据流向可以是双向的。2.pytorchvariable(用于封装tensor,便于自动求导的变量类型,在pytorch0.4.0之后版本已被并入tensor)基本属性:data,dtype,shape,device,requires_grad,is_leaf,grad,grad_fn3.is_leaf是否为叶子节点:用户创......
  • docker-compose version 版本匹配
     version:'3.8'services:rmqnamesrv:image:apache/rocketmq:5.1.0container_name:rmqnamesrvports:-9876:9876restart:alwaysprivileged:truevolumes:-/usr/local/rocketmq/nameserver/logs:/home/rocket......
  • 如何拆解Unity 2022.3版本的AssetBundle
    1)如何拆解Unity2022.3版本的AssetBundle2)Unity2022LTS版本的稳定性3)关于AssetBundle禁用TypeTree之后的一些可序列化的问题这是第363篇UWA技术知识分享的推送,精选了UWA社区的热门话题,涵盖了UWA问答、社区帖子等技术知识点,助力大家更全面地掌握和学习。UWA社区主页:community.......