手把手安装nerfstudio,包含tiny-cuda-nn手动编译安装步骤

一、nerfstudio简介

找到这里你大概也知道nerfstudio是做啥的了,这是官网地址,因复现实验经常使用到,在此记录安装过程。

Nerfstudio 最初是由伯克利人工智能研究中心(BAIR)KAIR 实验室的伯克利学生于 2022 年 10 月作为研究项目(论文)的一部分推出的开源项目。 Nerfstudio 提供了一个简单的应用程序接口,可简化创建、训练和测试 NeRF 的端到端流程。 该库通过将每个组件模块化,支持更易于解释的 NeRFs 实现。 有了更多模块化的 NeRF,在探索技术的过程中为用户创造更友好的体验。 这是一个对贡献者友好的资源库,目标是建立一个社区,让用户可以更轻松地在彼此贡献的基础上进行开发。在这里插入图片描述

二、安装

我在ubuntu20.04上安装的,要求python版本在3.8及以上,提前安装anaconda。

conda create --name nerfstudio -y python=3.8
conda activate nerfstudio
python -m pip install --upgrade pip

创建环境
接下来安装依赖,官方推荐使用torch版本 2.1.2 跟CUDA 11.8。如果你在当前环境中安装过torch可先卸载,没有可跳过。

pip uninstall torch torchvision functorch tinycudann

安装 PyTorch 2.1.2 和 CUDA 11.8,要是安装到一半失败的话多试几次,安装不了的话直接换镜像源。

pip install torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

在这里插入图片描述
要构建必要的 CUDA 扩展,还需要 cuda-toolkit。 建议使用 conda 安装(截图是已经安装成功的):

conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit

在这里插入图片描述
接下来安装tiny-cuda-nn

pip install ninja git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch

假如出现意外tiny-cuda-nn安装失败,建议直接从github拉取源码编译安装。

git clone https://github.com/NVlabs/tiny-cuda-nn

在这里插入图片描述
不出意外的话拉取的tiny-cuda-nn会缺少fmt的依赖,如果报fmt文件夹已存在,将空的fmt文件夹删除,再执行。

git clone https://github.com/fmtlib/fmt.git dedependencies/fmt

假如报错没得cutlass(不是cutclass哦),同fmt处理在这里插入图片描述

git clone https://github.com/NVIDIA/cutlass.git dependencies/cutlass

废了九牛二虎之力,终于安装上,太TM操蛋了。
在这里插入图片描述
tips:假如拉取项目太慢,可以直接到github下载压缩包解压到相应位置。

ok,到这里就完成依赖的安装,开始安装nerfstudio

pip install nerfstudio

在这里插入图片描述

如果你想从源代码编译安装最新版本可以如下操作:

git clone https://github.com/nerfstudio-project/nerfstudio.git
cd nerfstudio
pip install --upgrade pip setuptools
pip install -e .

以下是可选安装,可使nerfstudio更方便使用。(可以不用管)
当 CLI 发生变化(例如更新了 nerfstudio)时需要重新运行。

ns-install-cli

安装开发包

pip install -e .[dev]
pip install -e .[docs]

使用Pixi(一款基于现有conda生态系统的快速软件包管理器,可以快速启动开发环境)

curl -fsSL https://pixi.sh/install.sh | bash
git clone https://github.com/nerfstudio-project/nerfstudio.git
cd nerfstudio
pixi run post-install
pixi shell
要在Windows中安装tiny-cuda-nn,您需要按照以下步骤进行操作: 1. 首先,确保您的计算机已经安装了合适的CUDA驱动程序。您可以从NVIDIA官方网站下载并安装最新的CUDA驱动程序。 2. 在安装CUDA驱动程序后,您需要安装CMake工具。您可以从CMake官方网站下载并安装最新版本的CMake。 3. 接下来,您需要安装Git客户端,以便从GitHub上获取tiny-cuda-nn的源代码。您可以从Git官方网站下载并安装最新版本的Git。 4. 打开一个命令行终端或Git Bash终端,并导航到您想要存储tiny-cuda-nn源代码的目录。 5. 在命令行终端或Git Bash终端中,运行以下命令来克隆tiny-cuda-nn的GitHub存储库: ``` git clone https://github.com/tiny-cuda-nn/tiny-cuda-nn.git ``` 6. 克隆完成后,导航到tiny-cuda-nn源代码的目录。然后,创建一个新的文件夹来构建项目,比如"build": ``` cd tiny-cuda-nn mkdir build cd build ``` 7. 在构建文件夹中,运行以下命令使用CMake生成构建文件: ``` cmake .. ``` 8. 完成CMake配置后,运行以下命令来编译tiny-cuda-nn: ``` cmake --build . ``` 9. 编译完成后,您可以在构建文件夹中找到生成的可执行文件。您可以运行该可执行文件来使用tiny-cuda-nn。 请注意,上述步骤假设您已经正确安装CUDA驱动程序、CMake和Git,并且您的计算机满足tiny-cuda-nn的要求。如果在安装过程中遇到任何问题,请参考tiny-cuda-nn的文档或GitHub存储库中的说明。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值