5,322
edits
CUDA: Difference between revisions
→Linux
(→Linux) |
|||
| Line 17: | Line 17: | ||
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | ||
sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | ||
sudo apt update | sudo apt-get update | ||
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb | wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb | ||
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb | sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb | ||
sudo apt update | sudo apt-get update | ||
# Install NVIDIA driver | # Install NVIDIA driver | ||
sudo apt install - | sudo apt-get install --no-install-recommends nvidia-driver-450 | ||
# Reboot. Check that GPUs are visible using the command: nvidia-smi | # Reboot. Check that GPUs are visible using the command: nvidia-smi | ||
| Line 29: | Line 29: | ||
sudo apt-get install --no-install-recommends \ | sudo apt-get install --no-install-recommends \ | ||
cuda-10-1 \ | cuda-10-1 \ | ||
libcudnn7=7.6. | libcudnn7=7.6.5.32-1+cuda10.1 \ | ||
libcudnn7-dev=7.6. | libcudnn7-dev=7.6.5.32-1+cuda10.1 | ||
# Install TensorRT. Requires that libcudnn7 is installed above. | # Install TensorRT. Requires that libcudnn7 is installed above. | ||