Get Caffe2 Error when import the pytorch-lightning.
🐛 Bug
Get Caffe2 Error when import the pytorch-lightning.
To Reproduce
Steps to reproduce the behavior:
import pytorch_lightning
WARNING:root:This caffe2 python run does not have GPU support. Will run in CPU only mode.
WARNING:root:Debug message: /opt/caffe2/build/lib/libcaffe2.so: undefined symbol: dnnLRNCreateForward_F32
CRITICAL:root:Cannot load caffe2.python. Error: /opt/caffe2/build/caffe2/python/caffe2_pybind11_state.so: undefined symbol: _Py_ZeroStruct

Environment

CUDA:

GPU:

Tesla P100-PCIE-16GB


available:         True
version:           10.0.130


Packages:

numpy:             1.18.1
pytorch_lightning   0.7.5
pyTorch_debug:     False
pyTorch_version:   1.2.0
tensorboard:       2.2.1
tqdm:              4.45.0


System:

OS:                Linux
architecture:

64bit


processor:         x86_64
python:            3.7.6
version:           #76-Ubuntu SMP Wed Feb 12 03:02:44 UTC 2020