NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Other

3.14k stars 741 forks source link

issue when install flownet2.0 #258

Open Pixie8888 opened 3 years ago

Pixie8888 commented 3 years ago

Hi, when I running "bash install.sh", I encountered this problem: channelnorm_kernel.cu:3:10: fatal error: ATen/cuda/CUDAContext.h: No such file or directory

include <ATen/cuda/CUDAContext.h>

      ^~~~~~~~~~~~~~~~~~~~~~~~~

compilation terminated. error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1 Any idea how to solve it? My pytorch version is 0.4.1.post2

anandravi24 commented 2 years ago

I got that error initially too. The configuration which worked for me is: Conda Cuda 9 Python 3,6,2 pytorch 1.0.0