【三维重建】windows10情况设置tiny-cuda-nn具体教程

打印 上一主题 下一主题

主题 806|帖子 806|积分 2418

1. 前言

本人在复现water-splatting时,需要设置tiny-cuda-nn,与此同时,出现了很多问题,在此进行简朴概述。

2.安装Pytorch

情况版本要求保持一致:CUDA(物理机),Pytorch,CUDA Toolkit
   留意这里的tiny-cuda-nn对应的是真实的物理机安装的cuda版本号(情况变量里设置的cuda),不是假造情况下的cuda版本号,否则纵然tiny-cuda-nn安装乐成,也不能利用gpu。
  在此本人的CUDA为12.4,固然官方为11.8,但是不停报错,以是终极决定,安装12.4的CUDA



利用anaconda3新建tiny-cuda-nn假造情况:
  1. conda create -n water_splatting python=3.8
  2. activate water_splatting
复制代码
然后安装对应版本pytorch和cuda包:
  1. conda install pytorch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 pytorch-cuda=12.4 -c pytorch -c nvidia
复制代码
3.源码编译安装tiny-cuda-nn

这里源码githup的readme已经给出了差别操作系统下c++编译器和cuda的版本要求。(本人已经安装VS2022社区版,同时已经安装MVS,也在情况变量进行设置)

需要将D:\DevelopTools\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\14.42.34433\bin\Hostx64\x64加入情况变量,个人根据本身下载位置添加

在命令行中输入cl可以查看到下图,即可证明VS安装完毕且情况变量设置完毕

3.1git下载(不保举)

  1. # 可能需要科学上网(不推荐)
  2. git clone --recursive https://github.com/nvlabs/tiny-cuda-nn
复制代码
git下载太慢了,会时常中断导致部门文件缺失(dependencies\fmt和dependencies\cutlass文件夹内是不全),需要再单独下载再放到文件对应文件夹内。
3.2直接下载(保举)

【tiny-cuda-nn的下载地点】【fmt的下载地点】【cutlass的下载地点】
解压fmt和cutlass,并将文件内容放到dependencies\fmt和dependencies\cutlass中。

管理者身份打开"x64 Native Tools Command Prompt for VS 2022"终端,cd到符合的路径,输入以下指令:
  1. cd tiny-cuda-nn/bindings/torch
  2. activate water_splatting
  3. python setup.py install
复制代码
留意一定要以管理者身份打开,否则出现以下问题

实行过程中可能出现: “Error compiling objects for extension” ,需要修改tiny-cuda-nn\bindings\torch\setup.py中的内容:
  1. cmdclass={"build_ext": BuildExtension}
  2. # 修改为
  3. cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}
复制代码
乐成安装。

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

参考
【Windows情况下nerfstudio情况设置及复现(含tinycudann安装、poster数据)】

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

何小豆儿在此

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表