5,323
edits
(→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 | sudo apt 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 | sudo apt update | ||
# Install NVIDIA driver | # Install NVIDIA driver | ||
sudo apt | sudo apt install -y 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 | ||
# Install development and runtime libraries (~4GB) | # Install development and runtime libraries (~4GB) | ||
sudo apt | sudo apt install -y cuda-10-1 \ | ||
libcudnn7=7.6.4.38-1+cuda10.1 \ | |||
libcudnn7-dev=7.6.4.38-1+cuda10.1 | |||
# Install TensorRT. Requires that libcudnn7 is installed above. | # Install TensorRT. Requires that libcudnn7 is installed above. | ||
sudo apt | sudo apt install -y libnvinfer6=6.0.1-1+cuda10.1 \ | ||
libnvinfer-dev=6.0.1-1+cuda10.1 \ | libnvinfer-dev=6.0.1-1+cuda10.1 \ | ||
libnvinfer-plugin6=6.0.1-1+cuda10.1 | libnvinfer-plugin6=6.0.1-1+cuda10.1 |