Depending on the pytorch version used return_complex arg as well as profiler are not available. It’s a matter of removing those.
I asked a college to run it in another gpu and found the same issue for:
GPU 1: NVIDIA GeForce GTX 1080 Ti
GPU 2: NVIDIA TITAN Xp

PyTorch version: 1.7.0+cu110
Is debug build: True
CUDA used to build PyTorch: 11.0
ROCM used to build PyTorch: N/A
OS: Ubuntu 18.04.3 LTS (x86_64)
GCC version: (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
Clang version: Could not collect
CMake version: version 3.18.0
Python version: 3.6 (64-bit runtime)
Is CUDA available: True
CUDA runtime version: Could not collect
GPU models and configuration:
GPU 0: NVIDIA GeForce RTX 3090
GPU 1: NVIDIA GeForce GTX 1080 Ti
GPU 2: NVIDIA TITAN Xp
Nvidia driver version: 470.82.00
cuDNN version: Could not collect
HIP runtime version: N/A
MIOpen runtime version: N/A
Versions of relevant libraries:
[pip3] facenet-pytorch==2.5.1
[pip3] numpy==1.19.4
[pip3] torch==1.7.0+cu110
[pip3] torch-mir-eval==0.3.post2
[pip3] torchaudio==0.7.0
[pip3] torchsummary==1.5.1
[pip3] torchvision==0.8.1+cu110
[conda] Could not collect