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
打完收工