Pytorch-Correlation-extension
Pytorch-Correlation-extension copied to clipboard
Custom implementation of Corrleation Module
hi, there are something went wrong and i can't figure it out. python setup.py install running install running bdist_egg running egg_info writing Correlation_Module/spatial_correlation_sampler.egg-info/PKG-INFO writing dependency_links to Correlation_Module/spatial_correlation_sampler.egg-info/dependency_links.txt writing requirements to...
Hello, I find it get slower after using SpatialCorrelationSampler (from 0.647 to 0.832 to run a iteration). Do you know then possible problems? Maybe it is the cuda(11.4) or cudnn(8.2.4)...
Hello, I'm trying to make the following code work on Collab:https://github.com/ZhenboSong/mono_velocity/blob/master/train_crop_velocity.py But since I can't install their Correlation function. self.corr = Correlation(pad_size=md, kernel_size=1, max_displacement=md, stride1=1, stride2=1, corr_multiply=1) I found myself...
Hi, Thanks for this nice implementation, it brings convenient to our project. I have slightly modify the ```setup.py``` script to compile for multiple GPU_CCs. And I belief it would benefit...
OS :Linux Pytorch=1.8.0 torchversion=0.9.0 python 3.8 (torch18) cwc@imc-Z9PE-D8-WS:~/aot-benchmark-main$ pip install spatial_correlation_sampler Collecting spatial_correlation_sampler Using cached spatial_correlation_sampler-0.3.0.tar.gz (8.6 kB) Requirement already satisfied: torch>=1.1 in /home/cwc/anaconda3/envs/torch18/lib/python3.8/site-packages (from spatial_correlation_sampler) (1.8.0) Requirement already satisfied:...
I can't install the spatial-correlation-sampler on my remote gpu server. I have the following constraints: - Nvidia driver >= 460.27.03 - OS: Fedora release >= 32 Has someone successfully installed...
Hi, my env is: python 3.8 pytorch 1.8.0 cudatoolkit 11.4 torchvision 0.9.0 gcc 9.3.0 I tried commands: pip install spatial-correlation-sampler The result is: ``` (torch18) rongchang@BETA:~/extent_tools/Pytorch-Correlation-extension-master$ pip install spatial-correlation-sampler Defaulting...
I used the correlation_package used in Flownet2 of NVIDIA and the parameters are > self.corr = Correlation(pad_size=4, kernel_size=1, max_displacement=4, stride1=1, stride2=1, corr_multiply=1) and now I use spatial-correlation-sampler with the parameters:...
hi there, I encountered some errors,when I tried with "pip install spatial-correlation-sampler -vvv".Please help me. cuda version: 10.1.168 pytorch version:1.7.1 python version:3.7.0 gcc 7.5.0 [2.txt](https://github.com/ClementPinard/Pytorch-Correlation-extension/files/5762788/2.txt)
Hello: thanks for your sharing. I can use torch-cuda_101.1 normally. But when I import the spatial_correlation_sampler, ImportError: libcudart.so.9.2: cannot open shared object file: No such file or directory. Ps: I...