ubuntu安装深度学习环境(anaconda、cuda、cudnn)

在linux(ubuntu)下装配一整套深度学习环境

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

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

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
找到对应的版本下载后,使用sh安装

然后安装cuda

然后是安装你所需要版本的cuda和cudnn,我安装的是cuda_11.0和cudnn_8.0.1
过去版本的cuda下载地址
https://developer.nvidia.com/cuda-toolkit-archive
给自己当个备忘录吧,不然找起来又找不到
ubuntu安装深度学习环境(anaconda、cuda、cudnn)_第1张图片
ubuntu安装深度学习环境(anaconda、cuda、cudnn)_第2张图片
按照官网命令行安装
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链接
https://developer.nvidia.com/rdp/cudnn-archive

然后解压

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

如果得到下面的结果,则安装成功。
ubuntu安装深度学习环境(anaconda、cuda、cudnn)_第3张图片

参考博文:
https://blog.csdn.net/weixin_44383114/article/details/123648338

你可能感兴趣的:(ubuntu,深度学习,linux)