【tiny-cuda-nn的下载地址】
GitHub - NVlabs/tiny-cuda-nn:闪电般的C++/CUDA神经网络框架
【fmt的下载地址】
GitHub - fmtlib/fmt,位于 b0c8263cb26ea178d3a5df1b984e1a61ef578950
【cutlass的下载地址】
GitHub - NVIDIA/cutlass 在 1eb6355182a5124639ce9d3ff165732a94ed9a70
解压fmt和cutlass,并将文件内容完全辅助放到dependencies\fmt和dependencies\cutlass中。
安装VS 2019,以管理者身份打开"x64 Native Tools Command Prompt for VS 2019"终端,cd到合适的路径,输入以下指令:
cd tiny-cuda-nn/bindings/torch
conda activate tiny-cuda-nn(环境名称)
python setup.py install
等待编译成功就行
执行过程中可能出现: “Error compiling objects for extension” ,需要修改tiny-cuda-nn\bindings\torch\setup.py中的内容:
cmdclass={"build_ext": BuildExtension}
# 修改为
cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}
成功安装。
# 查看环境中的是否成功安装包
conda list
验证tiny-cuda-nn是否能使用
import tinycudann as tcnn
总结
尽可能简单、详细的介绍windows11环境配置tiny-cuda-nn详细教程。