do you think the Python version is causing a problem here since NVIDIA driver, PyTorch, and CUDA are same across three machines:
Ubuntu 20.04 working
Ubuntu 18.04 not working
CentOS 7 not working
details: File "pose_optimization.py", line 287, in find_optimal_pose vertices=torch.matmul(vertices.unsqueeze(0), rotations_init), RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)` · Issue #29 · facebookresearch/phosa · GitHub 3