minhocomcomai/tensorflow-pytorch-cuda Spec Ubuntu 18.04 CUDA 11.1.1 CUDNN 8.0.5 TensorRT 7.2.3 miniconda3 Python 3.8.5 PyTorch 1.8.1 Tensorflow 2.4.1