![从零开始学TensorFlow2.0](https://wfqqreader-1252317822.image.myqcloud.com/cover/188/33692188/b_33692188.jpg)
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人
2.2 在Linux系统中安装TensorFIow 2.0的GPU版本
在Linux系统中安装TensorFlow 2.0的GPU版本需要Nvidia显卡的支持。因此,需要安装cuda10。
(1)安装Nvidia驱动,需要安装410.48以上版本,禁止Ubuntu自带驱动,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_15_4.jpg?sign=1739296088-qJmH0GYDMJBEE5nnBKUx4qgLRGd4R2aN-0-9c0e6b5a1edd58748f93c146a17764ec)
(2)在打开的文件中添加如下代码。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_15_5.jpg?sign=1739296088-LpEvw9vNn9xQRnmFZJqbGBPg9WSLJppb-0-f294d21d870b72cff9cfc620202749f4)
(3)更新配置后验证,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_15_6.jpg?sign=1739296088-VCTHf3reXA4Qv5XPK4z1uKxlCk5vSDr9-0-1664db36a302e304a1e5c854cf08b821)
(4)安装Nvidia驱动,下载完成后执行如下命令。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_1.jpg?sign=1739296088-45QeoygmZRnt1kr8nXKdTERzFLuqS8wC-0-a2abbb4b4b324ecf916f42fc19f6209f)
(5)重新启动图形界面,查看显卡驱动,命令如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_2.jpg?sign=1739296088-kk08XNx5GC3lo21Vru8dcd3bLQl0IAOv-0-56efbc57343768ecd240b2ce508e35eb)
(6)cuda下载完成后进行安装,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_3.jpg?sign=1739296088-WDyfFMmWdl850ntebY6Y8ptl7XksFXW4-0-9cb0405863b52f23d6853d9fc5ba6774)
(7)安装完成后配置到动态链接库,操作如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_4.jpg?sign=1739296088-hGGf7i8p2DBOHuwZJzUOK98EFavtGnNq-0-d9e229af0a4073a5adc57482769b5107)
(8)添加如下语句。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_5.jpg?sign=1739296088-UwKYQqR77mP7Woh9vTwSDsSaYoBDvREH-0-fb98f949b6ef57893c0ab79856d2a04b)
(9)保存退出后,执行如下命令。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_6.jpg?sign=1739296088-TdjkSWCZQn24RQABVmfwlRmKa9Lg5s3S-0-d3b09db50cbe19829a9ad33451c84ea8)
(10)配置到环境变量,进行如下操作。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_7.jpg?sign=1739296088-4N2FuMknBBa2rq1u8v8WA8a0Ya0NLQlA-0-7d0e5e7a7bf38e0285c2d951475d441c)
(11)在文件末尾添加如下路径。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_8.jpg?sign=1739296088-JqrTPybBMqfNzJ1CZquJK96s6lXPMxZe-0-dc7ecfae385e56cc4e6b85beb084a808)
(12)运行如下命令使配置生效。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_9.jpg?sign=1739296088-oA6hL3RABXgFAmDBO1RUNTnrXogfHu5u-0-8bc5581d9c951e38f189197f9e6ff222)
(13)下载cuDNN后进行安装,操作如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_10.jpg?sign=1739296088-Yz6i2FaISycEkbUG2WnjLHK8umfiI39b-0-1dff5a09a76f943f6b5d0a1928cfc514)
(14)安装TensorFlow 2.0的GPU版本,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_16_11.jpg?sign=1739296088-hkgQmS1ybl0is9lXFz0iC4JF9wEQ1s8A-0-f56f2b672e5f033aa38ca28fe16a0adf)