1. 下载压缩包,解压到任意文件目录下

我已经踩好多坑咯,直接把未编译但完整的文件压缩放这里咯!(对了我的环境是py38_cu116_pyt1131)

https://download.csdn.net/download/weixin_47530040/88552806?spm=1001.2014.3001.5501


2. 在tiny-cuda-nn文件夹,使用cmake进行编译,依次运行下面命令

cd tiny-cuda-nn

cmake . -B build -DCMAKE_BUILD_TYPE=RelWithDebInfo

cmake --build build --config RelWithDebInfo -j

3. 编译完成后,再进行tiny-cuda-nn的安装:

cd bindings/torch

python setup.py install

okok,大功告成咯!!感谢csdn上记录自己踩坑经历的前辈们!!

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐