Hi @user152363, glad to hear that you got it working! Yes, you can run the pytorch-ssd repo on a GPU-enabled Linux PC or server that has PyTorch/CUDA/cuDNN installed on it. I run it on my Ubuntu laptop that has a GeForce card in it using the NGC PyTorch container (using this container all you need to have installed in Ubuntu is the NVIDIA driver and docker).