Linux (tested on Ubuntu 20.04.1 LTS) NVIDIA GPU + CUDA CuDNN (tested on RTX 3080 Ti) Please use PyTorch2.0.1 to avoid compilation errors (other versions should be ...