Computer vision container that includes Jupyter notebooks with built-in code hinting, Anaconda, CUDA 11.8, TensorRT inference accelerator for Tensor cores, CuPy (GPU drop in replacement for Numpy), PyTorch, PyTorch geometric for Graph Neural Networks, TF2, Tensorboard, and OpenCV for accelerated workloads on NVIDIA Tensor cores and GPUs.

opencv machine-learning computer-vision deep-learning image-processing pytorch vision tensorboard cupy tensorflow2 tensorrt-inference-accelerator

• Updated Oct 28, 2023
• Jupyter Notebook