Error during executing train.py

[JaehyeokJang]

Hi, I have some problem during executing train.py of e3d.

After I executed following commands of train.py

model = torch.nn.parallel.DistributedDataParallel(
    model.cuda(),
    device_ids=[dist.local_rank()],
    find_unused_parameters=True)

---

---

I got following error messages related to nvidia-modprobe.

/usr/bin/nvidia-modprobe: unrecognized option: "-f"

ERROR: Invalid commandline, please run /usr/bin/nvidia-modprobe --help
for usage information.

/usr/bin/nvidia-modprobe: unrecognized option: "-f"

ERROR: Invalid commandline, please run /usr/bin/nvidia-modprobe --help
for usage information.

Do you have any idea for solving this problem? Please check and let me know.

Thank you in advance for your help.

[JaehyeokJang]

FYI, current version of NVIDIA drivers, python and pytorch are as follows.

----------------------- NVIDIA -------------------------
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Wed_May__6_19:09:25_PDT_2020
Cuda compilation tools, release 11.0, V11.0.167
Build cuda_11.0_bu.TC445_37.28358933_0

Python : 3.7.7
Pytorch : 1.6.0

[zhijian-liu]

This seems to be related to PyTorch. Are you able to run other PyTorch training that uses distributed data-parallel?

[JaehyeokJang]

I got reference library from you and it is solved now. So I will close this issue. Thank you for your big effort!