中文網上能搜到的tensorflow-gpu安裝方法都很是麻煩,須要你看你的顯卡,看你的顯卡驅動,去nvidia官網本身下載cuda啥的,還不必定跟你最後安裝的tensorflow兼容。包括我也被這些辣雞文章害了,浪費一下午。最後發現直接用conda安裝就能自動匹配安裝對應的cuda等,還能隨意選擇版本,很是方便。python
進入官網,拉到最下面,根據你係統是64仍是32位下載安裝,通常win10都是64位。安裝就按默認選項就行。
網絡
打開Anaconda prompt,而後測試
conda create -n tensorflow python=3.7
spa
conda activate tensorflow
code
如下命令默認安裝符合你的驅動的最新版tensorflow:conda install tensorflow-gpu
若是想安裝指定版本,好比1.10.0,使用如下命令:conda install tensorflow-gpu=1.10.0
若是想安裝CPU版本,使用如下命令:conda install tensorflow
若是在安裝過程當中出現錯誤,通常是網絡問題,你們懂的,使用ke-xue衝浪get
首先輸入python
進入python交互環境,而後輸入如下代碼:it
import tensorflow as tf print(tf.__version__) print(tf.test.is_gpu_available())
預期結果應該以下圖,主要是最後輸出True,表示使用了GPU,這樣就表示安裝成功了。
io