第三篇 Ubuntu 20.04 搭建AI开发环境

Posted 2022-09-01 AI浩

摘要

这篇文章包括CUDA的安装方法、cuDNN安装方法、Anaconda的安装方法和Pycharm的安装方法。

CUDA安装方法

安装CUDA，我们先看一下主流的平台支持哪些CUDA的版本。首先是Pytoch，地址链接：

https://pytorch.org/get-started/locally/

我们看到Pytorch支持10.2、11.3、11.6版本。再看PaddlePaddle，地址：https://www.paddlepaddle.org.cn/

PaddlePaddle支持11.6、11.2、11.1、10.2、10.1版本。
通过上面的两个平台支持的CUDA版本，我们可以得出这两个平台都支持11.6，所以这次我们尝试用最新的11.6版本的CUDA。

在这里还要注意一点，10.2版本是长期版本，如果你的显卡是30系列以下的显卡，可以尝试用10.2版本。如果是30系列的显卡则只能使用11.X的版本。

1、检查驱动

输入命令：

nvidia-smi