search

ltkong218 / FastFlowNet

FastFlowNet: A Lightweight Network for Fast Optical Flow Estimation (ICRA 2021)
MIT License
261 stars 42 forks source link

TensorRT docker image with FastFlowNet #23

Closed pullmyleg closed 2 years ago

pullmyleg commented 2 years ago

Thanks for the repo!

In case other people are struggling to get a TensorRT environment working: https://hub.docker.com/r/pullmyleg/tensorrt8_cuda11.3_pytorch1.10.2_fastflownet

ltkong218 commented 2 years ago

Thanks for your contribution!

I have added your Docker TensorRT environment for FastFlowNet to the README.md file of this repository. Thanks!

zhanghanduo commented 2 years ago

I tried this docker but got an error with this line: import onnx_graphsurgeon as gs. When I tried to install onnx-graphsurgeon, it told me "ERROR: Could not find a version that satisfies the requirement onnx-graphsurgeon (from versions: 0.0.1.dev4, 0.0.1.dev5) ERROR: No matching distribution found for onnx-graphsurgeon" and " The package you are trying to install is only a placeholder project on PyPI.org repository. This package is hosted on NVIDIA Python Package Index.

This package can be installed as:

  $ pip install nvidia-pyindex
  $ pip install onnx-graphsurgeon

" Also I need password to install pip packages. It told me " permission denied".

pullmyleg commented 2 years ago

@zhanghanduo from memory I started with the base nvidia GPU docker container. Password for sudo will be: nvidia