温馨提示:如果你装CUDA是为了安装GPU版本的pytorch,那么其实不用看我下面关于CUDA安装的内容,因为你使用pytorch官网的安装命令,它会自动帮你安装相应的CUDA,但是cuDNN可能还得你手动安装。
-
安装英伟达专有驱动
利用ubuntu自带的附加驱动软件进行安装,打开附加驱动之后按下图进行操作
安装英伟达专有驱动.png
- 测试驱动是否安装成功以及查看驱动版本
打开终端输入nvidia-smi,查看输出情况。若驱动安装成功,会输出类似下图的结果。
下图中需要注意的有两点:标有1的方框里面显示的是当前安装的英伟达驱动版本号,标有2的方框里面显示的是当前驱动版本可以安装的CUDA最高版本号
测试驱动安装是否成功以及检查驱动版本.png
- 从英伟达官网下载对应的CUDA安装文件
由上一张图可知,驱动版本号为450.66,最高支持CUDA 11.0,所以点击CUDA 11.0对应的链接,进入下载页面。当然,也可以选择不安装支持的最高版本,安装其他版本。如果想知道当前安装的驱动能安装

本文详细介绍了如何在Ubuntu 20.04上安装英伟达专有驱动、CUDA 11.0以及cuDNN。首先通过附加驱动软件安装驱动,然后根据驱动版本从英伟达官网下载CUDA安装文件,并进行安装。安装过程中需注意取消驱动安装选项,最后设置环境变量并检查安装是否成功。安装cuDNN时,只需将下载的压缩包内容复制到CUDA安装目录相应位置。

8811

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



