5,323
edits
CUDA: Difference between revisions
→Linux
(→Linux) |
(→Linux) |
||
| Line 7: | Line 7: | ||
* Install [https://developer.nvidia.com/cuda-toolkit Cuda Toolkit] separately without the drivers.<br> | * Install [https://developer.nvidia.com/cuda-toolkit Cuda Toolkit] separately without the drivers.<br> | ||
** Use one of the deb install options. | ** Use one of the deb install options. | ||
* | * For machine learning, you may also want to install the following: | ||
** [https://developer.nvidia.com/rdp/cudnn-download | ** [https://developer.nvidia.com/rdp/cudnn-download cuDNN] | ||
** TensorRT | ** [https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing TensorRT] | ||
;Adapted from tensorflow | ;Adapted from [https://www.tensorflow.org/install/gpu#install_cuda_with_apt Tensorflow: Install cuda with apt] | ||
<pre> | <pre> | ||
# Add NVIDIA package repositories | # Add NVIDIA package repositories | ||
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | ||
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 | sudo apt install ./cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | ||
sudo apt-get 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 | ||