官网:https://docs.nerf.studio/quickstart/installation.html
1. 创建虚拟环境
当然,前提是已经安装了anaconda。
conda create --name nerfstudio -y python=3.8
conda activate nerfstudio
python -m pip install --upgrade pip
2. 安装pytorch
当然,前提是已经安装显卡驱动、cuda(11.8)和cudnn
pip install torch2.1.2+cu118 torchvision0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
踩坑:torch太大,下载不完整,安装失败
解决:复制下载whl链接,用迅雷先下载到本地,然后
pip install torch-2.1.2+cu118-cp38-cp38-win_amd64.whl
再安装torchvision,注意还是用上面那条完整安装命令(即pip install torch-2.1.2+cu118-cp38-cp38-win_amd64.whl torchvision),否则会卸载现有torch安装最新cpu版本torch
cuda-toolkit不安装也行
3. 安装tiny-cuda-nn
pip install ninja git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
踩坑:github连接不稳定,尤其代码中还包含子模块(其它代码)的情况
解决:将tiny-cuda-nn及其子模cutlass、fmt块都转移到gitee,然后在tiny-cuda-nn的.gitmodules中修改子模块链接。麻烦?我已完成以上工作,直接用我的仓库即可:
pip install ninja git+https://gitee.com/Xiaoshier2021/tiny-cuda-nn/#subdirectory=bindings/torch
4. 安装nerfstudio
pip install nerfstudio