1. 下载GPU驱动

去官网根据条件下载对应的GPU驱动文件,得到.run文件

2. 禁用Nouveau 驱动

禁用 Nouveau 驱动

3. 安装驱动(选择这一步,则在安装CUDA时可以不选择安装驱动;否则就需要选择安装CUDA时的驱动)

依次执行以下命令:

sudo apt install make
sudo apt install gcc
sudo chmod 777 xx.run
sudo apt-get remove --purge nvidia*
sudo bash xx.run

然后出现选择时,一路默认即可

最后验证是否安装成功,使用命令nvidia-smi,会出现以下证明成功了:
在这里插入图片描述

4. 安装CUDA

去官网下载相应版本的CUDA文件,文件格式为.run
在选择安装内容时,可以将驱动模块取消

sudo chmod 777 xx.run
sudo bash xx.run

安装完成后会出现如下信息:
在这里插入图片描述

意味着我们需要配置环境变量,配置环境变量相关教程参考:添加cuda到bashrc环境变量

安装成功后,使用命令nvcc -V查看CUDA版本:
在这里插入图片描述

5. 安装CUDNN

5.1 若是deb文件

官网下载对应版本的CUDNN,我下载的是deb版本,然后使用命令安装:sudo dpkg -i xxx.deb即可安装成功

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐