这里主要针对驱动已经安装好了的环境,如何安装合适的显卡驱动这里不多赘述,本文演示的系统为Ubuntu18.04,但理论上其他Linux发行版操作类似。
驱动适配
通过命令行输入nvidia-smi查看自己的显卡驱动版本以及支持的最大CUDA版本,下图第一行就显示了这些信息,可以看到,最大支持CCUDA10.2,更高版本的CUDA需要升级驱动程序。
安装包下载
CUDA的安装可以访问官网开发者工具中找到,链接给出,此时显示的结果如下,这是最新的CUDA11,点击右侧的红框选择历史版本。
找到CUDA10.2,按照需求进行选择,然后安装指示使用wget或者浏览器下载runfile文件。
CUDNN的下载也类似,链接也给出,这个下载需要注册开发者账号(简单几步注