# Cambricon PyTorch Model Migration Report
## Cambricon PyTorch Changes
| No. |  File  |  Description  |
| 1 | driver.py:3 | add "import torch_mlu" |
| 2 | driver.py:53 | change "self.device = torch.device('cuda', self.local_rank)" to "self.device = torch.device('mlu', self.local_rank) " |
| 3 | driver.py:162 | change "if torch.cuda.device_count() > 1:" to "if torch.mlu.device_count() > 1: " |
| 4 | driver.py:165 | change "torch.distributed.init_process_group(backend="nccl")" to "torch.distributed.init_process_group(backend="cncl") " |
| 5 | driver.py:166 | change "torch.cuda.set_device(local_rank)" to "torch.mlu.set_device(local_rank) " |
| 6 | lstm.py:1 | add "import torch_mlu" |
| 7 | hour_dataset.py:7 | add "import torch_mlu" |