pytorch import error

Today when I was trying the new version of torchvision, an error occurred:

1

ImportError: /usr/local/lib/libcaffe2_gpu.so: undefined symbol: _ZTIN6caffe210DAGNetBaseE

Some said it was a problem with conda or pip installation, but source installation is time consuming. Then I found this post(https://blog.csdn.net/daydayjump/article/details/84894443), which said one should comment this line:

1

export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

In order to explicitly get rid of this problem, I use:

1

export LD_LIBRARY_PATH=/usr/local/cuda/lib

All good!

The new torchvision has included lots of interesting CNN based models, e.g. Faster-RCNN, Mask-RCNN.