从零开始学TensorFlow2.0
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人

2.2 在Linux系统中安装TensorFIow 2.0的GPU版本

在Linux系统中安装TensorFlow 2.0的GPU版本需要Nvidia显卡的支持。因此,需要安装cuda10。

(1)安装Nvidia驱动,需要安装410.48以上版本,禁止Ubuntu自带驱动,代码如下。

(2)在打开的文件中添加如下代码。

(3)更新配置后验证,代码如下。

(4)安装Nvidia驱动,下载完成后执行如下命令。

(5)重新启动图形界面,查看显卡驱动,命令如下。

(6)cuda下载完成后进行安装,代码如下。

(7)安装完成后配置到动态链接库,操作如下。

(8)添加如下语句。

(9)保存退出后,执行如下命令。

(10)配置到环境变量,进行如下操作。

(11)在文件末尾添加如下路径。

(12)运行如下命令使配置生效。

(13)下载cuDNN后进行安装,操作如下。

(14)安装TensorFlow 2.0的GPU版本,代码如下。