I am trying to use Pytorch with Horovod. I am trying to run one of the example. I am getting the following error:
Traceback (most recent call last):
File "pytorch_imagenet_resnet50.py", line 3, in <module>
import torch
File "/ccs/home/amalik/.conda/envs/py366/lib/python3.6/site-packages/torch/__init__.py", line 84, in <module>
from torch._C import *
ImportError: /ccs/home/amalik/.conda/envs/py366/lib/python3.6/site-packages/torch/lib/libtorch_python.so: undefined symbol: MPIX_Query_cuda_support
However, when I import torch on simple python command, I don’t get the error.
a>> python
>>> import torch
>>>
no error message