ubuntu 18.04 安装cuda 10.0+cudnn 7.6

conda安装cuda和cudnn的一些可能问题

用conda维护虚拟环境时，可通过conda来安装cuda和cudnn，但在ubuntu下，安装了tensorRT后，编译sample时，会发生找不到cuda文件的问题。

# conda install cuda and cudnn
conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/ -y
conda install cudnn=7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/ -y

故此、用官方文件安装cuda和cudnn是另一个选择。

官方文件安装cuda和cudnn是另一个选择：

安装cuda

下载安装包

https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=clusterlocal

安装cuda

切换到安装文件所在目录，

1. Run `sudo sh ./cuda_10.0.130_410.48_linux.run`
2. Follow the command-line prompts
3. （安装补丁：sudo sh cuda_10.0.130.1_linux.run）

注意，如果已经安装了显卡驱动，不要重复安装，可能会带来问题。 (可选择不安装samples)

如需要建立软链接：

sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda

设置环境变量,保存退出

>sudo vim /etc/profile

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


>source /etc/profile

查看是否安装成功：

驱动：返回如下则成功

 cat /proc/driver/nvidia/version

cuda：返回如下则成功

nvcc -V

测试sample （optional）

切换到samples路径，如下、则成功

> cd /usr/local/cuda-10.0/samples/
> sudo make

# about 10min later

> cd bin/x86_64/linux/release

# 执行：
> ./deviceQuery

 

安装cudnn

下载安装包
https://developer.nvidia.com/rdp/cudnn-archive

解压安装包：

> cp cudnn-10.0-linux-x64-v7.6.4.38.solitairetheme8  cudnn-10.0-linux-x64-v7.6.4.38.tgz
> tar -xvf cudnn-10.0-linux-x64-v7.6.4.38.tgz

将头文件和lib文件拷贝：

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*

 

完工！