I have a GPU setup for Tensorflow 2.6.0. Cuda v11.2 and Cudnn v8.1.0 Enviroments paths are added. After that I started learning Unity ML Agents. I followed the installations instructions. I didn't get any Cudnn errors. So I assumed my GPU installation is also done, since it says PyTorch 1.7.1+cu110 But I couldn't train on GPU. I tried both --torch-device=cuda and --torch-device=cuda:0 parameters. What would be the proper GPU installation?