name: gaussian_splatting
channels:
- pytorch
- conda-forge
- defaults
dependencies:
- cudatoolkit=11.6
- plyfile
- python=3.7.13
- pip=22.3.1
- pytorch=1.12.1
- torchaudio=0.12.1
- torchvision=0.13.1
- tqdm
- pip:
- submodules/diff-gaussian-rasterization
- submodules/simple-knn
- opencv-python
- joblib
해당 코드를 이용해서 환경을 구성했는데, 자꾸 submodules 다운로드 과정에서 오류가 발생했음
export CUDA_HOME=/usr/local/cuda
export TORCH_CUDA_ARCH_LIST="8.6"
export FORCE_CUDA=1
이 코드를 쳐주니 다운로드가 완료되었음
'Code > Paper Open Source Codes' 카테고리의 다른 글
[Automatic 1111] 스테이블 디퓨전 (Stable-Diffusion) -webui 맥 Mac M1/M2/M3 로컬 설치 및 실행 (0) | 2024.03.19 |
---|---|
[runwayml] Stable-Diffusion (Official) 코드 실행 (0) | 2024.03.13 |