初始化环境
- 使用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
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!