月度归档: 2026 年 10 月

1 篇文章

在Linux上配置CUDA-Toolkit,以及装好torch相关库
我真傻,真的,我单知道做ai用python和torch就够了… 翻rwkv文档的时候知道这个模型可能会用到一些自定义算子(?)但是直到导入它的AutoTokenizer,程序突然Warning: 这怎么跟CUDA有关了……后来Claude大爹给我指出这是CUDA编译时找不到头文件了,这可能会导致fla/Triton相关算子会回退CPU。据此记录这篇…