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

ubuntu 18.04 安装cuda 10.0+cudnn 7.6_第1张图片

安装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)

ubuntu 18.04 安装cuda 10.0+cudnn 7.6_第2张图片

如需要建立软链接:

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

ubuntu 18.04 安装cuda 10.0+cudnn 7.6_第3张图片

 

安装cudnn

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

ubuntu 18.04 安装cuda 10.0+cudnn 7.6_第4张图片

解压安装包:

> 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*

 

完工!

 

你可能感兴趣的:(平台维护,深度学习)