当前位置： 首页 > article >正文

BUG解决：安装问题transformer_engine+pytorch

article 2025/1/23 12:07:59

安装命令

pip install transformer_engine[pytorch]

或

pip install git+https://github.com/NVIDIA/TransformerEngine.git@main

Cmake报错信息解决

如果出现cmake相关CUDA/CUDNN错误：
（eg.缺少cudnn.h）
更改CUDA_PATH和CUDNN_PATH

版本检查CUDA 12.0+ and cuDNN 8.1+.

首先找到环境下合适的cuda 版本，比如12.1

之后

vim ~/.bashrc

export PATH=/usr/local/cuda-12.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH
export CUDA_PATH=/usr/local/cuda-12.1
export CUDA_HOME=/usr/local/cuda-12.1
export CUDNN_PATH=/usr/local/cuda/include

ESC键 + :wq + 回车键

source ~/.bashrc

nvcc -V

http://www.kler.cn/a/514868.html

相关文章：

  • GPT 结束语设计 以nanogpt为例
  • Linux内核中的InfiniBand核心驱动：verbs.c分析
  • 【深度学习】2.视觉问题与得分函数
  • 深度学习中Batch Normalization(BN)原理、作用浅析
  • 将 AzureBlob 的日志通过 Azure Event Hubs 发给 Elasticsearch（1.标准版）
  • kotlin的协程的基础概念
  • 基于springboot+vue的高校社团管理系统的设计与实现
  • docker ubuntu:20.04构建c++ grpc环境
  • es的date类型字段按照原生格式进行分组聚合
  • QILSTE H13-320B2W高亮白光LED灯珠 发光二极管LED
  • 如何使用CRM数据分析和洞察来支持业务决策和市场营销？
  • 开源鸿蒙开发者社区记录
  • 深入了解 Java split() 方法：分割字符串的利器
  • AI时代的网络安全：传统技术的落寞与新机遇
  • Kubernetes入门学习
  • Spring Boot 事件驱动：构建灵活可扩展的应用
  • PostgreSQL 初中级认证可以一起学吗？
  • Servlet快速入门
  • Spring Boot MyBatis Plus 版本兼容问题（记录）
  • HOW - 基于master的a分支和基于a的b分支合流问题
  • Echarts现成案例
  • 阻燃高温尼龙行业：市场潜力巨大，引领材料科学新变革
  • AI 辅助 Codebase 本地工程检索
  • Linux内核中 Netfilter 框架的用户态工具iptables(配置防火墙规则)
  • Vue | 搭建第一个Vue项目（安装node,vue-cli）
  • ubuntu中xrandr多屏幕设置