/home/user4/anaconda3/lib/python3.7/site-packages/torch/include/ATen/cuda/CUDAContext.h:7:10: fatal error: cublas_v2.h: No such file or directory
#include <cublas_v2.h>
^~~~~~~~~~~~~
compilation terminated.
error: command '/usr/bin/nvcc' failed with exit status 1
更换Linux的GCC版本