Jump to content

CUDA: Difference between revisions

Line 11: Line 11:
** TensorRT
** TensorRT


;Copied from tensorflow
;Adapted from tensorflow
<pre>
<pre>
# Add NVIDIA package repositories
# Add NVIDIA package repositories
Line 23: Line 23:


# Install NVIDIA driver
# Install NVIDIA driver
sudo apt-get install --no-install-recommends nvidia-driver-430
sudo apt-get install --no-install-recommends nvidia-driver-440
# Reboot. Check that GPUs are visible using the command: nvidia-smi
# Reboot. Check that GPUs are visible using the command: nvidia-smi