matheustguimaraes

Install CUDA 10.0 and cuDNN v7.4.2 on Ubuntu 16.04

Originally posted on: https://matheustguimaraes.com/blog/cuda-cudnn-ubuntu-installation

Install the latest NVIDIA driver

Update package lists, download and install NVIDIA driver

sudo apt-get update
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt install nvidia-410

ortegatron

from detectron2.engine.hooks import HookBase
from detectron2.evaluation import inference_context
from detectron2.utils.logger import log_every_n_seconds
from detectron2.data import DatasetMapper, build_detection_test_loader
import detectron2.utils.comm as comm
import torch
import time
import datetime

class LossEvalHook(HookBase):