fidler-lab/polyrnn-pp-pytorch

Docker file for this project

qinliuliuqin opened this issue · 0 comments

Dear authors,

Thanks for releasing the code! Would it be possible for you to provide a docker file for efficiently running your code?
Just like this project did: ritm_interactive_segmentation.
I am now trying to dockerize your code, but fail to get the results from the web browser.
My Docker file is shown below:
"FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04

RUN apt-get update && apt-get install -y
build-essential
git
curl
firefox
nano

RUN apt-get install -y
virtualenv
python2.7
libxrender1
libsm6
libxext6
"
But every time I draw the bounding box on the web browser, I get the error:
"RuntimeError: CUDNN_STATUS_EXECUTION_FAILED"

Thank you!