Caused by: org.apache.spark.api.python.PythonException: Traceback (most recent call last):


 "/mnt/resource/hadoop/yarn/local/usercache/nsusshuser/appcache/application_1526397916826_0020/container_e01_1526397916826_0020_01_000002/pyspark.zip/pyspark/worker.py", line 166, in main

func, profiler, deserializer, serializer = read_command(pickleSer, infile)


 "/mnt/resource/hadoop/yarn/local/usercache/nsusshuser/appcache/application_1526397916826_0020/container_e01_1526397916826_0020_01_000002/pyspark.zip/pyspark/worker.py", line 55, in read_command

command = serializer._read_with_length(file)


 "/mnt/resource/hadoop/yarn/local/usercache/nsusshuser/appcache/application_1526397916826_0020/container_e01_1526397916826_0020_01_000002/pyspark.zip/pyspark/serializers.py", line 169, in _read_with_length

return self.loads(obj)


 "/mnt/resource/hadoop/yarn/local/usercache/nsusshuser/appcache/application_1526397916826_0020/container_e01_1526397916826_0020_01_000002/pyspark.zip/pyspark/serializers.py", line 455, in loads

return pickle.loads(obj, encoding=encoding)
ImportError: No module named 'ml'

Traceback (most recent call last):


 "/workspaces/speech/scripts/train.py", line 66, in train

loss.backward()


 "/opt/conda/lib/python3.6/site-packages/torch/tensor.py", line 166, in backward

torch.autograd.backward(self, gradient, retain_graph, create_graph)


 "/opt/conda/lib/python3.6/site-packages/torch/autograd/__init__.py", line 99, in backward

allow_unreachable=True)
# allow_unreachable flag
RuntimeError: expected device cpu but got device cuda:0
