issue when install flownet2.0

[Pixie8888]

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]

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