chantat

@geyyer I tried removing `CUB_DIR` like what you mentioned but I still run into the exact same errors. Please let me know if any extra information would be helpful for...

@MohammadMahdiJavanmard Yes, `import torch` works. We compile pytorch from source using the following commands: ``` pip3.7 install -r requirements.txt \ && TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" \ CMAKE_INCLUDE_PATH=${CAFFE_ROOT}/build/src...