上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版本,代码如下。