Requirement issue

[Rexime]

I tried everything, but just couldn't get it work. Could someone provide a working version of requirement.txt, please?

[gaodalii]

Here is the Dockerfile I used.
Most Python library versions are explicitly specified, it may be helpful as a reference for setting up the environment.

FROM nvcr.io/nvidia/cuda:12.1.1-devel-ubuntu22.04

ENV NVIDIA_DRIVER_CAPABILITIES=all

ENV TZ=Asia/Shanghai
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ 

RUN mv /etc/apt/sources.list /etc/apt/sources.list.bak && \
    echo "deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse" > /etc/apt/sources.list && \
    echo "deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb http://security.ubuntu.com/ubuntu/ jammy-security main restricted universe multiverse" >> /etc/apt/sources.list

# python
RUN apt-get update && \
    apt-get install -y software-properties-common && \
    apt-get update && \
    add-apt-repository -y ppa:deadsnakes/ppa && \
    apt-get update && \
    apt-get purge -y --auto-remove python3 && \
    apt-get install -y python3.10 python3.10-dev python3.10-distutils wget && \
    update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1

# pip
RUN wget https://bootstrap.pypa.io/get-pip.py && \
    python get-pip.py
        
# torch
RUN pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121  

# opencv-python
RUN apt-get update && apt-get -y install libgl1 libglib2.0-0 && \
    pip install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python

# 
RUN apt-get update && apt-get install -y build-essential ninja-build git htop curl

# nvdiffrast
RUN pip install --no-cache-dir 'git+https://github.com/NVlabs/nvdiffrast/'   
    
# kaolin
RUN pip install kaolin==0.17.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.1.0_cu121.html

#
RUN pip install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple \
    'huggingface-hub==0.25.2' \
    'einops==0.8.0' \
    'omegaconf==2.3.0' \
    'torchmetrics==1.4.3' \
    'trimesh==4.5.0' \
    'xatlas==0.0.9' \
    'tensorboard' \
    'wandb' \
    'viser' \
    'setuptools==69.5.1' \
    'numpy < 2' \
    'typer'

# transformers pytorch-lightning segment-anything
WORKDIR /src
 
RUN git clone --branch v4.44.2 --single-branch --depth 1 https://github.com/huggingface/transformers.git && \
    cd transformers && \
    pip install -e . --no-build-isolation 

RUN git clone --branch 2.4.0 --single-branch --depth 1 https://github.com/Lightning-AI/pytorch-lightning.git && \
    cd pytorch-lightning && \
    pip install -e . --no-build-isolation 

RUN git clone --branch main --single-branch --depth 1 https://github.com/facebookresearch/segment-anything.git && \
    cd segment-anything && \
    pip install -e . --no-build-isolation