Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,790 Bytes
cf92dec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
# Note: conda dependencies have only 1 "="
# pip dependencies have 2 "=="
# Fuse pip dependencies together under one " - pip" item
# Otherwise, only some of the are installed, because conda creates a temporary requirements.txt file
# only the last -pip section
name: p3dmm
channels:
- pytorch
- conda-forge
- defaults
dependencies:
- python=3.9
- pip
- jupyter
# CUDA and PyTorch
- gcc<12 # Needs to be <12 because nvcc does not like gcc>11
- gxx
- torchvision
-
- nvidia/label/cuda-11.8.0::cuda-nvcc # for nvcc
- nvidia/label/cuda-11.8.0::cuda-cccl
- nvidia/label/cuda-11.8.0::cuda-cudart
- nvidia/label/cuda-11.8.0::cuda-cudart-dev # for cuda_runtime.h
- nvidia/label/cuda-11.8.0::libcusparse
- nvidia/label/cuda-11.8.0::libcusparse-dev
- nvidia/label/cuda-11.8.0::libcublas
- nvidia/label/cuda-11.8.0::libcublas-dev
- nvidia/label/cuda-11.8.0::libcurand
- nvidia/label/cuda-11.8.0::libcurand-dev
- nvidia/label/cuda-11.8.0::libcusolver
- nvidia/label/cuda-11.8.0::libcusolver-dev
- pip:
- pip:
- --extra-index-url https://download.pytorch.org/whl/cu118
- torch==2.7+cu118
- torchvision==0.22+cu118
- tyro
- environs
- omegaconf
- dreifus
- wandb
- pytorch_lightning
- opencv-python
- tensorboard
- wandb
- scikit-image
- pyvista
- chumpy
- h5py
- einops
- ninja
- mediapy
- face-alignment==1.3.3
- numpy==1.23
- git+https://github.com/facebookresearch/pytorch3d.git@stable
- git+https://github.com/NVlabs/nvdiffrast.git
# for MICA
- insightface
- onnxruntime
- loguru
- yacs
# facer
- distinctipy
- validators
- timm
|