gcc/g++,CUDA,CUDA ToolKit,Pytorch配置环境配置gaussian-splatting

gaussian-splatting

git clone需要递归,否则没有子模块

在创建conda环境前,需要检查c++是否可以调用glm库,如果无法

#include<glm/glm.hpp>

随便在网上下一个glm的文件夹放到 /usr/include/ 下。否则无法 pip install

然后

根据我的个人环境,选择了

gcc 11 + CUDA 11.8 + 相应pytorch

CUDA

pytorch之类的内容都写在了environment.yml里面,原本的环境文件需要修改,用以下这个就行

name: mygs
channels:
  - pytorch
  - conda-forge
  - defaults
  - nvidia
dependencies:
  - cudatoolkit=11.8
  - plyfile
  - python=3.10
  - pip=22.3.1
  - pytorch=2.2.0
  - torchaudio=2.2.0
  - torchvision=0.17.0
  - pytorch-cuda=11.8
  - tqdm
  - pip:
    - submodules/diff-gaussian-rasterization
    - submodules/simple-knn
posted @ 2024-05-28 03:48  千陌  阅读(47)  评论(0编辑  收藏  举报