Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.
And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.
Is CUDA 11.6 supported?
See original GitHub issueThis is a really promising project, congratulations and thanks for releasing it!
I’m trying to run the test script with your Eth3d model and this command:
python3 test.py --model_path path_to_mge_model --left img/test/left.png --right img/test/right.png --size 1024x1536 --output disparity.png
But the code hangs up and doesn’t return from this line in extractor.py:82:
self.conv2 = M.Conv2d(128, output_dim, kernel_size=1)
which is called form load_model in test.py:15
model = Model(max_disp=256, mixed_precision=False, test_mode=True)
My GPU is NVIDIA RTX A6000 and the CUDA version on the system is v11.6
Issue Analytics
- State:
- Created a year ago
- Comments:14
Top Results From Across the Web
Top Related Medium Post
Top Related StackOverflow Question
Troubleshoot Live Code
Top Related Reddit Thread
Top Related Hackernoon Post
Top Related Tweet
Top Related Dev.to Post
Top Related Hashnode Post
I have not tried training, but I was able to run with CUDA 11.3 in my Pytorch implementation, in case someone needs it: https://github.com/ibaiGorordo/CREStereo-Pytorch
based on @ibaiGorordo’s code, I enabled the training process with pytorch, here it is https://github.com/Qjizhi/CREStereo-Pytorch, thanks both @ibaiGorordo and the authors of this paper.