Model training failed "OSError: libcudart.so.8.0: cannot open shared object file"

[no1asian]

There's an error in the model training phase.

root@82a0db5cc7c4:~/ecs-deep-learning-workshop/mxnet/example/image-classification# python3 train_mnist.py --lr-factor 1
Traceback (most recent call last):
File "train_mnist.py", line 25, in
from common import find_mxnet, fit
File "/root/ecs-deep-learning-workshop/mxnet/example/image-classification/common/find_mxnet.py", line 20, in
import mxnet as mx
File "/usr/local/lib/python3.5/dist-packages/mxnet/init.py", line 24, in
from .context import Context, current_context, cpu, gpu, cpu_pinned
File "/usr/local/lib/python3.5/dist-packages/mxnet/context.py", line 24, in
from .base import classproperty, with_metaclass, _MXClassPropertyMetaClass
File "/usr/local/lib/python3.5/dist-packages/mxnet/base.py", line 213, in
_LIB = _load_lib()
File "/usr/local/lib/python3.5/dist-packages/mxnet/base.py", line 204, in _load_lib
lib = ctypes.CDLL(lib_path[0], ctypes.RTLD_LOCAL)
File "/usr/lib/python3.5/ctypes/init.py", line 347, in init
self._handle = _dlopen(self._name, mode)
OSError: libcudart.so.8.0: cannot open shared object file: No such file or directory