深度学习入门教程-ubuntu18.04安装CUDA 10

深度学习入门教程-安装CUDA 10

说明：

介绍如何安装CUDA 10

环境：

系统：ubuntu 18.04
显卡：GTX1080Ti

安装PPA：

安装依赖：

$ sudo apt-get install linux-headers-$(uname -r)

进入官方找到合适版本:https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/
我们选择这个cuda-repo-ubuntu1804_10.1.168-1_amd64.deb版本
安装：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.168-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub 
sudo apt update
sudo dpkg -i cuda-repo-ubuntu1804_10.1.168-1_amd64.deb

sudo apt update
sudo apt install -y cuda

增加路径

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

启用环境

source ~/.bashrc

检测：

执行命令：

nvcc --version

效果如下：

ubuntu@AiDLHost:~$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105

查看cuda版本也可以执行

cat /usr/local/cuda/version.txt

参考：