windows 11 本地运行ER-NeRF及pytorch3D安装

在Windows 11上顺利运行ER-NeRF需要正确配置环境。首先创建一个名为ernerf的Python 3.10虚拟环境,接着安装特定版本的PyTorch和 Torchvision。由于系统存在多个CUDA版本,需在环境变量中切换到CUDA 11.3。然后,通过pip安装requirements.txt列出的依赖,并解决可能出现的版本冲突。安装PyTorch3D前,要确保编译环境就绪,包括安装nvidiacub并设置CUB_HOME环境变量。此外,还需安装CUB的其他依赖,最后在虚拟环境中配置Visual Studio环境变量,并安装tensorflow-gpu 2.8.0,可能需要调整protobuf版本以避免冲突。
摘要由CSDN通过智能技术生成

ER-NeRF本地运行只要梳理好依赖版本,运行起来就很顺畅

  1. conda create -n ernerf python=3.10
    创建本项目虚拟环境
    
  2. conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
    若windows有多个版本的cuda,需要在环境变量中切换至cuda 11.3,主要是系统变量中的CUDA_PATH,NVCUDASAMPLES_ROOT,PATH进行编辑为自己需要的cuda版本
    
  3. pip install -r requirements.txt
    会遇到部分依赖包版本冲突,根据实际情况处理即可
    
    
  4. pip install "git+https://github.com/facebookresearch/pytorch3d.git"
    安装pytorch3d之前需先安装好pytorch3d的编译环境,如下:
    
    conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
    conda install -c fvcore -c iopath -c conda-forge fvcore iopath
    因本环境采用的cuda版本是11.3,所以需手动安装 nvidiacub,并设置系统环境变量 CUB_HOME,指向位置为CMakeLists.txt所在路径,安装方式如下:
    
  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值