tinny-cuda-cu安装

初始化环境

  1. 使用anaconda3新建tiny-cuda-nn虚拟环境:
conda create -n tiny-cuda-nn python=3.8
activate tiny-cuda-nn
  1. 安装对应版本pytorch和cuda包:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

源码编译安装tiny-cuda-nn

  1. 直接下载
    tiny-cuda-nn, fmt, cutlass;
  2. 移动文件夹
    将fmt和cutlass文件内容移动到dependencies\fmt和dependencies\cutlass中(在tinny-cuda-cu)路径下。
  3. 编译
cd tiny-cuda-nn/bindings/torch
activate tiny-cuda-nn
python setup.py install
  1. 验证tiny-cuda-nn是否能使用
import tinycudann as tcnn

加入其他库文件

conda env update -f environment.yaml
posted @ 2024-01-28 20:33  阿奘  阅读(388)  评论(0编辑  收藏  举报