标签：cutlass nn tinny tiny cuda cu
初始化环境
- 使用anaconda3新建tiny-cuda-nn虚拟环境：
conda create -n tiny-cuda-nn python=3.8
activate tiny-cuda-nn
- 安装对应版本pytorch和cuda包：
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
源码编译安装tiny-cuda-nn
- 直接下载
tiny-cuda-nn, fmt, cutlass;
- 移动文件夹
将fmt和cutlass文件内容移动到dependencies\fmt和dependencies\cutlass中（在tinny-cuda-cu）路径下。
- 编译
cd tiny-cuda-nn/bindings/torch
activate tiny-cuda-nn
python setup.py install
- 验证tiny-cuda-nn是否能使用
import tinycudann as tcnn
加入其他库文件
conda env update -f environment.yaml
标签：cutlass,
nn,
tinny,
tiny,
cuda,
cu
From： https://www.cnblogs.com/fydream/p/17993289