Se você possui uma placa gráfica NVIDIA compatível e os drivers instalados no Linux, veja como encontrar a versão do NVIDIA CUDA no Linux.
NVIDIA vs AMD no Linux: Qual GPU você deve escolher?
GPU Linux: descubra a melhor escolha entre NVIDIA e AMD para seu sistema!