当前位置: 首页 > news >正文

torch版本应该跟cuda、cudacnn的版本一致

今天执行脚本:
from transformers import HfArgumentParser
莫名奇妙就报了:
ImportError: libcudnn.so.9: cannot open shared object file: No such file or directory

实际上是对整个llm架构不了解:
+-----------------------------+
|        transformers         |  ← 你直接调用的库
| (BertModel, GPT2Tokenizer)  |
+-----------------------------+↓
+-----------------------------+
|           torch             |  ← 实际执行计算的引擎
| (Tensor, nn.Module, Adam)   |
+-----------------------------+↓
+-----------------------------+
| CUDA / cuDNN / CPU / MPS    |  ← 硬件加速(GPU/Metal等)
+-----------------------------+j
即使transformers的版本对上了,那如果torch和cuda的版本没对上,还是不行。
所以把已经成功的环境保存一下,以便查阅:
torch                    2.3.1+cu118
torchaudio               2.3.1+cu118
torchvision              0.18.1+cu118
transformers             4.44.2
cuda            11.8                    /data/cuda/cuda-11.8/cuda  
cudnn           v8.8.1.3+cuda-11.8      /data/cuda/cuda-11.8/cudnn/v8.8.1.3   
nccl            v2.15.1+cuda-11.8       /data/cuda/cuda-11.8/nccl/v2.15.1      
tensorrt        v8.6.0.12+cuda-11.8     /data/cuda/cuda-11.8/tensorrt/v8.6.0.12 

  


 
http://www.wxhsa.cn/company.asp?id=384

相关文章:

  • 小白如何零成本搭建一个属于自己的私人知识库
  • 安装mysql数据库,从下载到配置的详细教程
  • 根据端口找到进程id
  • 双因子验证网站(aspsms.com/en/registration/)无法注册——Capcha Error
  • MathType7下载安装2025最新下载+安装教程(附安装包)
  • mysql导入数据库,从基础命令到高效技巧
  • 基于“北斗+卫星互联网”的低空飞行服务保障基础设施
  • [BJOI2018] 染色 题解
  • 【完结10章】Java大模型工程能力必修课,LangChain4j 入门到实践
  • CVE-2025-30208 Vite开发服务器任意文件读取漏洞
  • Claude Code 从入门到精通:最全配置指南和工具推荐
  • 故障分析:11GR DATAGRUAD环境BROKER配置Fast-Start Failover
  • Cesium Shader内置变量 czm_*
  • IDA Pro 9.2 发布 - 强大的反汇编程序、反编译器和多功能调试器
  • 传统
  • Java 那些基础又关键的事儿
  • 2025-09-10
  • Codeforces Round 1047 (Div. 3)
  • sentinel-1.8.0 安装
  • 数据结构与算法-27.树-并查集
  • wpf XAML设计器在加载用户控件的时候,提示null引用等直接执行了用户控件里构造函数代码的问题
  • 设计模式-策略
  • Linux中怎么调整系统inode数量?
  • DARPA AI网络挑战赛技术框架全解析:自动化漏洞挖掘与修复系统构建
  • 数据库基本查询语句
  • 【项目实战】基于WS63的鸿蒙星闪红外遥控车(循迹、超声波避障、远程控制、星闪/红外遥控)有教程代码
  • macbook pro怎么安装windows系统
  • XSS与CSRF的联系与区别
  • 异或
  • apche 2.4 开启mod_cache_disk和mod_deflate后,磁盘上缓存的是压缩后的文件