0
点赞
收藏
分享

微信扫一扫

ubuntu安装CUDA和CUDNN,简洁版

独兜曲 2022-03-12 阅读 85

 

一、准备工作

Nvidia驱动的安装,看我另外一个文章ubuntu18.04 安装Nvidia驱动的三种方式(必看)_低吟浅笑-CSDN博客

打开终端,输入

nvidia-smi

出来下列信息:

 出来该信息,说明驱动无误。查看驱动显示出来的CUDA版本,所安装的版本尽量与这个接近,驱动可以向下兼容,也就是可以安装比显示出来的CUDA版本低的版本。如图所示

二、安装CUDA

1. 官网下载相应的版本,需要考虑nvidia驱动:CUDA Toolkit Archive | NVIDIA Developer

2. 根据 nvidia驱动的版本下载cuda,注意下面有个 wget 网址的选项,如果wget下载较慢,可以打开链接直接下载。

 3.  安装 cuda

进入下载好的文件夹,输入

sudo sh cuda_10.0.130_410.48_linux.run

注意,新版的cuda安装是个小窗口,需要按 Enter 选择哪个不安装,旧版是通过提问的方式来的。

第一个选项选 accept

除了安装 驱动 的选项,其他都选 yes

4.配置环境变量

输入下面的命令打开bashrc

sudo vi ~/.bashrc
或者
sudo gedit ~/.bashrc

在最下面添加下面内容,注意 里面的 cuda-10.0 需要换成自己的版本,自己可以去 /usr/local/ 下面看一下呢

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

重新启动配置文件

source ~/.bashrc

三、安装CUDNN

举报

相关推荐

0 条评论