Linux (CentOS) 使用 Tensorflow-gpu (1080ti)

搞了一陣子,總算能在 Linux 上成功使用 Tensorflow GPU 版本
簡單紀錄一下 --

LAB 機器是 X299 + 1080ti,沒有 onboard gpu

這導致 Ubuntu 一個尷尬的問題 -- 太舊的版本 splash 會有問題無法顯示,太新的版本無法使用舊的 CUDA

幸好 CentOS 7 可以解決上述問題,所以不用強求一定要使用 Ubuntu,可以解決問題就先頂著用

另外,CUDA 一樣要使用 9.0,9.2 或是 10 都不行,不用浪費時間測試了

================================================
首先下載 Driver


使用 nvidia-smi 測試一下,有列出 GPU 表示成功


繼續下載 CUDA 9.0,下面有兩份 Guide 可以參考


下載 cudnn for cuda 9.0 (要註冊)


下載 Anaconda 並安裝

這邊要注意,新的 Conda 使用的 Python 是 3.7,安裝後要把 Python 改用 3.6,沒注意到版本差異卡了一些時間

conda install python=3.6



最後安裝 Tensorflow 並測試,可以用 virtualenv 方式


全流程安裝參考文件
install Tensorflow with GPU support on Centos 7


打完收工

這個網誌中的熱門文章

GAN 入門筆記

CNN 進階:RCNN、YOLO、SSD