Нужно установить драйвер Nvindia + Cuda

```
 sudo apt install nvidia-cuda-toolkit
 ```