- 博客(5)
- 资源 (1)
- 收藏
- 关注
原创 Windows环境下flash-attention安装
机器学习里Transformer模型大量使用flash-attention这个自注意力模块,多次安装遇到一堆问题,希望能给大家一点帮助。结合了多篇前人经验。
2024-02-13 23:25:23 2390 2
原创 PyTorch安装避坑
如果没有安装过cuDNN需要根据另外教程安装,此处我以前安装过,故省略。)中选择正确选项,然后在你需要安装的环境中复制“Run this Command:”后方的内容并回车键执行。默认指令(如下)安装的是CPU版本PyTorch,要调用显卡核心必须安装对应版本PyTorch。起因是更新torch版本后,运行程序会发现无法调用CUDA,走的时候CPU而非GPU。如果此前安装过,请先进行卸载,否则如果版本号相同会认为已经安装过,安装前首先检查自己的CUDA版本,在命令行中输入下方命令。在PyTorch下载页(
2024-02-13 16:45:30 653
原创 SpringDoc最新版极速入门
由于使用了高版本的SpringBoot(>=3.0),年久失修的Springfox已经无法正常配置与兼容新版本的SpringBoot,因此换用了新的Swagger。
2023-08-12 12:34:15 283
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人