【深度学习】【三维重建】windows10环境配置tiny-cuda-nn详细教程

【深度学习】【三维重建】windows10环境配置tiny-cuda-nn详细教程

文章目录


前言

本人windows10下使用【Instant Neural Surface Reconstruction】算法时需要搭建tiny-cuda-nn环境,故此以详细教程以该算法依赖的环境版本为参照。

很多读者反馈在win10下出现很多问题,博主现在就重新出一篇win10版的博文,博文在博主结尾分享了读者所需的下载链接。

win11下的朋友可以直接参考【win11下配置tiny-cuda-nn参考】。

确定版本对应关系

windows10环境配置详细教程】,
环境版本要求保持一致:CUDA(物理机),Pytorch,CUDA Toolkit

注意这里的tiny-cuda-nn对应的是真实的物理机安装的cuda版本号(环境变量里配置的cuda),不是虚拟环境下的cuda版本号,否则即使tiny-cuda-nn安装成功,也不能使用gpu。

博主的cuda版本新装了11.8,可以参考【安装多版本CUDA及其切换】

安装GPU版本的pytorch教程,pytorch-gpu版本需要根据个人计算机去安装相应版本。

本人的安装版本为:

Python 3.8.7 + Pytorch 2.0.1 + CUDA Toolkit 11.8

使用anaconda3新建tiny-cuda-nn虚拟环境:

bash 复制代码
conda create -n tiny-cuda-nn python=3.8
activate tiny-cuda-nn

然后安装对应版本pytorch和cuda包:

bash 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

源码编译安装tiny-cuda-nn

这里源码githup的readme已经给出了不同操作系统下c++编译器和cuda的版本要求。

git下载(不推荐)

bash 复制代码
# 可能需要科学上网(不推荐)
git clone --recursive https://github.com/nvlabs/tiny-cuda-nn

git下载太慢了,会时常中断导致部分文件缺失(dependencies\fmt和dependencies\cutlass文件夹内是不全),需要再单独下载再放到文件对应文件夹内。

直接下载(推荐)

tiny-cuda-nn的下载地址】【fmt的下载地址】【cutlass的下载地址

解压fmt和cutlass,并将文件内容放到dependencies\fmt和dependencies\cutlass中。

安装VS 2019,以管理者身份打开"x64 Native Tools Command Prompt for VS 2019"终端,cd到合适的路径,输入以下指令:

bash 复制代码
cd tiny-cuda-nn/bindings/torch
activate tiny-cuda-nn
python setup.py install

执行过程中可能出现: "Error compiling objects for extension" ,需要修改tiny-cuda-nn\bindings\torch\setup.py中的内容:

python 复制代码
cmdclass={"build_ext": BuildExtension}
# 修改为
cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}

成功安装。

bash 复制代码
# 查看环境中的是否成功安装包
conda list

验证tiny-cuda-nn是否能使用

python 复制代码
import tinycudann as tcnn

总结

尽可能简单、详细的介绍windows10环境配置tiny-cuda-nn详细教程。

博主这里分享俩个下载链接,一个是tiny-cuda-nn的完整包【百度云下载,提取码:xgyt】,读者可以下载后直接用它安装到自己的虚拟环境中;另一个是带有tiny-cuda-nn的anaconda虚拟环境【百度云下载,提取码:l7di】,读者可以直接使用虚拟环境,不再需要额外安装tiny-cuda-nn,cuda版本是11.8。

相关推荐
昨日之日20065 分钟前
Wan2.2-S2V - 音频驱动图像生成电影级质量的数字人视频 ComfyUI工作流 支持50系显卡 一键整合包下载
人工智能·音视频
SEO_juper3 小时前
大型语言模型SEO(LLM SEO)完全手册:驾驭搜索新范式
人工智能·语言模型·自然语言处理·chatgpt·llm·seo·数字营销
攻城狮7号3 小时前
腾讯混元翻译模型Hunyuan-MT-7B开源,先前拿了30个冠军
人工智能·hunyuan-mt-7b·腾讯混元翻译模型·30个冠军
zezexihaha3 小时前
从“帮写文案”到“管生活”:个人AI工具的边界在哪?
人工智能
算家云3 小时前
nano banana官方最强Prompt模板来了!六大场景模板详解
人工智能·谷歌·ai大模型·算家云·ai生图·租算力,到算家云·nano banana 提示词
暴躁的大熊3 小时前
AI助力决策:告别生活与工作中的纠结,明析抉择引领明智选择
人工智能
Gyoku Mint4 小时前
提示词工程(Prompt Engineering)的崛起——为什么“会写Prompt”成了新技能?
人工智能·pytorch·深度学习·神经网络·语言模型·自然语言处理·nlp
梁小憨憨4 小时前
zotero扩容
人工智能·笔记
大数据张老师4 小时前
AI架构师的思维方式与架构设计原则
人工智能·架构师·ai架构·后端架构
AKAMAI4 小时前
Entity Digital Sports 降低成本并快速扩展
人工智能·云计算