安装环境

我的是cuda11.8及对应pytorch,这个影响不是很大.VS2019 社区版本特别注意,是VS2019 社区版本,我发布的时候,我试了2017,2022都不行,注意下我文章的时间。

安装流程

下载源码,注意要使用git下拉下来,不然依赖库要一个一个下。

git clone --recursive https://github.com/nvlabs/tiny-cuda-nn  

依赖库在tiny-cuda-nn/dependencies可以看看下面的依赖文件夹是否有文件。
windows使用ninja很难,所以注意去除ninja。
在bindings\torch下的setup.py文件最后两行左右可以看到代码

cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}
Logo

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

更多推荐