说明:
nvidia 30系显卡仅支持cuda11.0及以上版本,对应cudnn最低版本为8.0,tf版本为2.4.0
在win系统中无法实现30系显卡运行tf1的代码
该教程使用的环境如下:
- Ubuntu20.04
- 3060 显卡
- cuda 11.1
- cudnn 8.0.5
- python 3.6
- tensorflow 1.15
- 其中 python 版本和 tensorflow 版本是固定的
- 简陋的目录
- 安装显卡驱动
- 安装cuda
- 安装cudnn
- 安装python(直接略过,我使用的conda 里面创建了一个python3.6的虚拟环境)
- 安装tensorflow1.15
安装 显卡驱动
# 依次执行以下命令
ubuntu-drivers devices #查看电脑所需要的驱动
# 通过上述命令也可能返回为空什么都不显示,添加官方ppa的源,在更新以下
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo ubuntu-drivers autoinstall #让其自动安装显卡驱动
#安装完成后通过nvdia-smi仍然无法查到显卡型号。重启一下。
#通过设置->软件更新->附加驱动里面查看选择相应的显卡驱动
安装 CUDA
- 在nvidia官网下载相应cuda文件
- cuda文件链接
- 选择相应版本进行安装