CUDA: Difference between revisions

CUDA (view source)

109 bytes removed , 6 February 2021

5,323

edits

@@ Line 24: / Line 24: @@
 # Install NVIDIA driver
-sudo apt-get install --no-install-recommends nvidia-driver-460
+sudo apt install nvidia-driver-460
 # Reboot. Check that GPUs are visible using the command: nvidia-smi
+sudo apt install cuda
 # Install development and runtime libraries
-sudo apt-get install --no-install-recommends \
+sudo apt install libcudnn8 libcudnn8-dev
-    cuda libcudnn7 libcudnn7-dev
+#sudo apt-get install -y libnvinfer6 libnvinfer-dev libnvinfer-plugin6
-# Install TensorRT. Requires that libcudnn7 is installed above.
-sudo apt-get install -y libnvinfer6 libnvinfer-dev libnvinfer-plugin6
 </syntaxhighlight>