error: command 'gcc' failed with exit status 1

[avani17101]

I followed https://github.com/hyangwinter/flownet3d_pytorch/issues/8 and replaced AT_CHECK and THCState_getCurrentStream(state) with TORCH_CHECK and at::cuda::getCurrentCUDAStream() respectively. But I get this error while running python setup.py install

System: OS: Ubuntu 20.04 LTS Python version: 3.7.6 PyTorch version: 1.8.1+cu102 Torchvision version: 0.9.1 CUDA/cuDNN version: 10.0/7

[whiterabbitfollow]

@avani17101

I followed the steps in #8 and it worked fine.

From the screenshot you have provided it looks like you are using getCurrentCUDAStream wrong. Instead try:

cudaStream_t stream = at::cuda::getCurrentCUDAStream();

My changes in "ball_query.cpp"

Hope it helps!