ubuntu安装深度学习环境（anaconda、cuda、cudnn）

ubuntu安装深度学习环境（anaconda、cuda、cudnn）

在linux（ubuntu）下装配一整套深度学习环境

如何在ubuntu上装深度学习环境？

首先，下载Anaconda3，在清华源上下载要快很多

/
找到对应的版本下载后，使用sh安装

然后安装cuda

然后是安装你所需要版本的cuda和cudnn，我安装的是cuda_11.0和cudnn_8.0.1
过去版本的cuda下载地址

给自己当个备忘录吧，不然找起来又找不到


按照官网命令行安装
accept
有驱动的把它带的驱动用空格取消，然后install
安装好之后，配置环境变量。用下面的命令打开.bashrc

sudo vim ~/.bashrc

按 I键进入vim的插入编辑模式，在最下面插入以下内容。为自己环境下可以生效的环境变量配置

export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda/bin:$PATH"
export CUDA_HOME="/usr/local/cuda:$CUDA_HOME"

编辑完成后，按 esc键退出vim编辑器，再输入 :wq保存文件并退出。然后，输入以下命令，激活更新后的环境变量。

source ~/.bashrc

1
注意，上面路径中是用/cuda而不是 /cuda-11.0，因为接下来需要通过软链接，以实现多个CUDA版本共存。输入下面代码，即可完成软链接的生成，其中 /cuda-11.0替换成自己的cuda安装目录名称。

sudo rm -rf /usr/local/cuda  #删除之前生成的软链接
sudo ln -s /usr/local/cuda-11.0 /usr/local/cuda  #生成新的软链接

最后，用以下命令，查看当前cuda版本。

nvcc -V
nvcc -version

接下来安装cudnn

备忘录，cudnn链接

然后解压

tar -xzvf 压缩包名称

再执行以下命令

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

执行下面的命令，查看是否安装成功

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

如果得到下面的结果，则安装成功。

参考博文：