TK1 python comport caffe error

Question

Traceback (most recent call last):
File "/home/ubuntu/python_data_test.py", line 9, in
import caffe
File "/home/ubuntu/caffe/python/caffe/__init__.py", line 1, in
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
File "/home/ubuntu/caffe/python/caffe/pycaffe.py", line 13, in
from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ImportError: libcudart.so.6.5: cannot open shared object file: No such file or directory