目录
一. 安装NVIDIA驱动
1. 首先,查看NVIDIA驱动是否已安装
2.安装NVIDIA驱动
二.安装cuda
1. 查看NVIDIA版本对应的cuda版本
2. 选择CUDA版本(上限原则)
3. 安装cuda
4. 安装cndnn
三.安装tensorflow、pytorch深度学习框架
1. 安装tensorflow-gpu
2. 安装pytorch
四.参考
一. 安装NVIDIA驱动
1. 首先,查看NVIDIA驱动是否已安装
在终端输入以下命令,如果显示以下结果,说明NVIDIA驱动已经正常安装,否则,需要重新安装;
nvidia-smi
2.安装NVIDIA驱动
方法一:使用Ubuntu自带的镜像源安装
第一步:先在软件中搜索“软件和更新”,然后选择一个可用的镜像源,我选择的是清华的镜像源。我的选择如下:
第二步:选择源之后,还是在软件更新中,选择附加驱动,然后选择合适的驱动,应用更改就好,我本机选择的是nvidia-driver-455的驱动。
第三步:完成安装后,在终端输入 nvidia-smi,如果可以显示以下结果,恭喜你,安装NVIDIA驱动成功啦。
方法二:下载NVIDIA官方驱动
第一步: 进入NVIDIA官方下载与之匹配的驱动: https://www.nvidia.cn/geforce/drivers/ ,下图是我本机驱动匹配的信息。点击“开始搜索”,进入NVIDIA驱动下载界面。