When running code that imports PyTorch in the Robotarium environment, the import fails before any user code is executed.
import torch
File "/usr/local/lib/python3.10/site-packages/torch/__init__.py", line 431, in <module>
from torch._C import * # noqa: F403
ImportError: /usr/local/lib/python3.10/site-packages/torch/lib/libtorch_cuda.so: undefined symbol: ncclCommShrink
When running code that imports PyTorch in the Robotarium environment, the import fails before any user code is executed.