【NERF】Ubuntu20.04系统从0开始搭建nerf-pytorch环境

本文详细指导了如何安装NVIDIA驱动、Anaconda、CUDA、CUDNN以及nerf-pytorch所需的Python环境,包括创建虚拟环境、安装PyTorch和其他相关库,最后演示了如何运行示例DEMO。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

目录

安装nvidia驱动

安装anaconda

安装CUDA

安装CUDNN

克隆nerf-pytorch代码

创建并激活虚拟环境

安装PyTorch

安装其他的库

运行DEMO


安装nvidia驱动

打开软件与更新

我选的是阿里源,也可以选清华源,都可以的。

选择适合自己的版本,先进行应用更改在进行重新启动。

然后就可以啦!真的太方便辽!

安装anaconda

  • 下载安装包

 anaconda下载官网:Free Download | Anaconda 

wget https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh
  • 运行安装
sh Anaconda3-2023.07-2-Linux-x86_64.sh

按enter键;输入yes按enter键;

此处回车选择默认在用户主目录下创建一个名为anaconda3的文件夹作为安装地址。

  • 输入conda -V命令(注意conda与-V之间有一个空格)可查看当前Anaconda版本
conda -V
  • 输入python命令查看是否可以使用python
python

安装CUDA

  • 下载安装包

CUDA下载地址: CUDA Toolkit Archive | NVIDIA Developer

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run

  • 运行安装
sh cuda_11.3.0_465.19.01_linux.run

更改安装路径

(因为我们是一个服务器多个人用,所以要安装在自己的账户下)

  • 配置并更新环境变量
vim ~/.profile

#添加
export PATH="/home/lyc/cuda-11.3/bin:$PATH"
export LD_LIBRARY_PATH="/home/lyc/cuda-11.3/lib64:/home/lyc/cuda-11.3/mylib/lib64:$LD_LIBRARY_PATH"

source ~/.profile
  •  输入nvcc -V命令(注意nvcc与-V之间有一个空格)可查看当前CUDA版本

安装CUDNN

  • 下载安装包

cudnn下载地址:cuDNN Archive | NVIDIA Developer

  • 解压
xz -d cudnn-linux-x86_64-8.9.4.25_cuda11-archive.tar.xz
tar -xvf cudnn-linux-x86_64-8.9.4.25_cuda11-archive.tar
  • 将include和lib中的内容复制到cuda文件夹中
cp include/cudnn.h /home/lyc/cuda-11.3/include
cp lib/libcudnn* /home/lyc/cuda-11.3/lib64
  •  添加读取权限
chmod a+r /home/lyc/cuda-11.3/include/cudnn.h /home/lyc/cuda-11.3/lib64/libcudnn*

克隆nerf-pytorch代码

nerf-pytorch代码网址:https://github.com/yenchenlin/nerf-pytorch

git clone https://github.com/yenchenlin/nerf-pytorch.git

创建并激活虚拟环境

conda create -n nerf python=3.7
conda activate nerf

安装PyTorch

PyTorch官网:Start Locally |PyTorch

conda install 或者 pip install 都可以,二选一就行。 

#二选一即可

# CUDA 11.3
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

# CUDA 11.3
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

验证pytorch是否安装成功:

安装其他的库

根据requirements.txt,用pip安装其他的库:

pip install imageio
pip install imageio-ffmpeg
pip install matplotlib
pip install configargparse
pip install tensorboard
pip install tqdm
pip install opencv-python

运行DEMO

  • 下载 lego 和 fern数据集 
bash download_example_data.sh

  • 训练一个低分辨率的 lego NeRF
python run_nerf.py --config configs/lego.txt

运行成功咯:

训练结果存放目录:logs/blender_paper_lego。

训练50000次:

训练100000次:

训练150000次:

训练200000次:

NERF PyTorch是一个用PyTorch重新实现的NERF(神经辐射场)库。你可以在GitHub上找到源码地址:https://github.com/yenchenlin/nerf-pytorch。这个实现相比原始回购速度提高了4-7倍,目前仅支持“blender”数据类型,但未来将添加更多格式和培训选项。 在NERF PyTorch中,有一个名为run_nerf_helpers.py的文件,其中包含一个名为Embedder的类。这个类用于进行位置编码,并采用了一些超参数进行初始化。在create_embedding_fn方法中,它根据设置的参数创建了一个嵌入函数列表,这些函数将用于对输入进行编码。最后,embed方法使用这些嵌入函数对输入进行编码,返回编码后的结果。 总而言之,NERF PyTorch是一个用PyTorch重新实现的NERF库,它提供了更快速的实现,并支持位置编码和嵌入。你可以在GitHub上的源码地址中找到更多详细信息。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [4、nerfpytorch)](https://blog.csdn.net/weixin_50973728/article/details/126048095)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* [NERF_pytorch:pytorch重新实现NERF](https://download.csdn.net/download/weixin_42131276/15662042)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值