基于Ubuntu 18.04系统
一、安装Nvidia驱动
使用显卡的基础,直接使用ubuntu仓库安装
1. 首先查看显卡可用的驱动
ubuntu-drivers devices
2. 安装推荐驱动
sudo apt install nvidia-driver-470-server
3. 查看安装是否成功
nvidia-smi
参考:Ubuntu下安装NVIDIA驱动的三种方法_Willard Leozi的博客-优快云博客_ubuntunvidia驱动安装官方 GeForce 驱动程序 | NVIDIA
二、 CUDA环境安装
多数AI框架通过cuda api 来使用英伟达显卡进行高性能计算,cuda开发者官网。

本文详细介绍了在基于Ubuntu 18.04系统上安装NVIDIA显卡驱动,CUDA环境,以及cuDNN库的步骤,包括检查驱动、安装推荐驱动、CUDA版本选择与安装、cuDNN下载与配置,并提供了一些关键操作的测试方法和注意事项。
最低0.47元/天 解锁文章
1544

被折叠的 条评论
为什么被折叠?



