- 從零開始學TensorFlow2.0
- 趙銘 歐鐵軍編著
- 304字
- 2020-09-25 12:47:40
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版本,代碼如下。
