- 博客(6)
- 收藏
- 关注
RSS订阅原创 Triton学习-01-vector-add
我选择的是直接根据官方示例的源码进行学习链接：首先使用源码进行执行的时候，遇到了第一个报错：AttributeError: 'BangDriver' object has no attribute 'get_active_torch_device'查阅相关的信息发现有网友说是NVIDIA的相关接口变掉了，导致triton中无法加载(不是特别确定，因为我使用的是国产的MLU显卡)，总之这个网友给了一个解决方案，是可以采纳使用的。将替换为这样我们就获取到了当前正在使用的GPU名称和卡号。
2025-07-11 09:56:24
585
原创 MLU370M8 Glip适配
写下载数据集脚本 touch download.sh。执行下载，需要等待一些时间(两三个小时)Flickr30K数据集自行前往。下载Objects365数据集。
2024-09-05 21:25:05
369
原创 寒武纪MLU370M8 stable_diffusion适配
修改 ldm/moudles/encoders/moudles.py 100行 将version路径改成自己的模型路径。注释modules/launch_utils.py 410-414 和 421-423行。注释modules/launch_utils.py 398-400行。修改modules/sd_vae_approx.py 57行。注释modules/devices.py 83行。模型拷贝到自己项目路径。
2024-09-05 21:11:04
1145
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人