Originally posted on: https://matheustguimaraes.com/blog/cuda-cudnn-ubuntu-installation
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
| 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): |
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