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 @   阿奘  阅读(902)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
  1. 1 黄昏 手嶌葵
黄昏 - 手嶌葵
00:00 / 00:00
An audio error has occurred.
点击右上角即可分享
微信分享提示