NVlabs / FB-BEV

Official PyTorch implementation of FB-BEV & FB-OCC - Forward-backward view transformation for vision-centric autonomous driving perception
Other
608 stars 40 forks source link

Error when Evaluation #34

Closed MarioUncle-Wang closed 2 months ago

MarioUncle-Wang commented 3 months ago

Hello, I have completed the model training phase , but an error occurred during the evaluation. The specific information is as follows: 1714092298627

heng-yuwen commented 2 months ago

same here

heng-yuwen commented 2 months ago

Upgrade your mmcv-full to 1.5.3

MarioUncle-Wang commented 2 months ago

Upgrade your mmcv-full to 1.5.3

i have already upgarde my mmcv-full to 1.5.3 but it doesnt work. my pip lists are as follows: 屏幕截图 2024-05-07 10:02:02 屏幕截图 2024-05-07 10:02:19

heng-yuwen commented 2 months ago

Upgrade your mmcv-full to 1.5.3

i have already upgarde my mmcv-full to 1.5.3 but it doesnt work. my pip lists are as follows: 屏幕截图 2024-05-07 10:02:02 屏幕截图 2024-05-07 10:02:19

My working Dockerfile as below:

FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04

# install python3.8 and env packages
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update && apt-get install wget -yq \
    && apt-get install build-essential g++ gcc -y \
    && apt-get install libgl1-mesa-glx libglib2.0-0 -y \
    && apt-get install openmpi-bin openmpi-common libopenmpi-dev libgtk2.0-dev git -y \
    && apt-get install python3.8-dev tmux zip unzip -y \
    && apt-get clean && rm -rf /var/lib/apt/lists/*

RUN ln -s /usr/bin/python3.8 /usr/bin/python
RUN rm /usr/bin/python3 && ln -s /usr/bin/python3.8 /usr/bin/python3

RUN wget https://bootstrap.pypa.io/get-pip.py && python ./get-pip.py && rm get-pip.py
RUN pip install --upgrade pip

RUN pip install --no-cache-dir pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
RUN pip install --no-cache-dir mmcv-full==1.5.3 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.12.0/index.html
RUN pip install --no-cache-dir mmdet==2.24.0 mmsegmentation==0.24.0 timm nuscenes-devkit==1.1.6

# install FB-BEV
WORKDIR /root
RUN git clone https://github.com/NVlabs/FB-BEV.git
WORKDIR /root/FB-BEV
ENV PYTHONPATH=${PYTHONPATH}:/root/FB-BEV

RUN pip install --no-cache-dir -e .
RUN pip install --no-cache-dir numpy==1.20.3 spconv-cu113

# download ckpts
RUN mkdir ckpts && cd ckpts && \
    wget https://github.com/zhiqi-li/storage/releases/download/v1.0/fbocc-r50-cbgs_depth_16f_16x4_20e.pth
heng-yuwen commented 2 months ago

@MarioUncle-Wang try mmcv-full==1.6.0.

MarioUncle-Wang commented 2 months ago

@MarioUncle-Wang try mmcv-full==1.6.0.

it worked! Thanks for your support.