movienet / movienet-tools

Tools for movie and video research
http://movienet.github.io
271 stars 31 forks source link

RTX3070+CUDA11.3+pytorch1.11.0 #31

Open LeoLee321 opened 2 years ago

LeoLee321 commented 2 years ago

Will the project support 30 series graphics cards? CUDA10.2+pytorch1.5 is unavailable for 30 graphics cards. I am looking forward to supporting the 30 series, thanks!

bulatnv commented 1 year ago

Hello. I is this library is not deprecated, the needed support for CUDA >= 11.4 versions.

Build with CUDA 11.4 gives error: Dockerfile to reproduce:

FROM nvcr.io/nvidia/pytorch:22.09-py3
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y ffmpeg python3-opencv
RUN pip3 install numpy
RUN pip3 install opencv-python==4.5.5.64
RUN pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
RUN pip3 install Cython
RUN git clone https://github.com/movienet/movienet-tools && cd movienet-tools && python -m pip install -e .