Open umairjavaid opened 3 years ago
Looks like this error is related to this: https://github.com/pytorch/pytorch/issues/61011. Did you use CUDA 11.4? If so, please retry with CUDA 11.0.
I have three directories: cuda, cuda-11 and cuda 11.4 on my system. I am giving the path to cuda-11 in my case.
The error does not occur with CUDA v11.0, so there must be some problem with your CUDA setting (perhaps missing environment variables?). You can check the log to see which version of CUDA you are using. The build log prints the following information at the beginning:
...
...
...
-- USE_CUDA : ON
-- CUDA static link : OFF
-- USE_CUDNN : ON
-- CUDA version : 11.0
-- cuDNN version : 8.0.5
-- CUDA root directory : /home/gyeongin/cuda-11.0
...
...
...
I added the following to caffe2/utils/math_gpu.cu
#include <thrust/host_vector.h>
and it worked like charm.