windows最簡單三行命令安裝tensorflow-gpu版(使用anaconda)無需本身安裝cuda

中文網上能搜到的tensorflow-gpu安裝方法都很是麻煩,須要你看你的顯卡,看你的顯卡驅動,去nvidia官網本身下載cuda啥的,還不必定跟你最後安裝的tensorflow兼容。包括我也被這些辣雞文章害了,浪費一下午。最後發現直接用conda安裝就能自動匹配安裝對應的cuda等,還能隨意選擇版本,很是方便。python

1.安裝anaconda

進入官網,拉到最下面,根據你係統是64仍是32位下載安裝,通常win10都是64位。安裝就按默認選項就行。
image.png網絡

2.安裝tensorflow

打開Anaconda prompt,而後測試

1.建立tensorflow虛擬環境

conda create -n tensorflow python=3.7spa

2.激活環境

conda activate tensorflowcode

3.安裝tensorflow

如下命令默認安裝符合你的驅動的最新版tensorflow:
conda install tensorflow-gpu
若是想安裝指定版本,好比1.10.0,使用如下命令:
conda install tensorflow-gpu=1.10.0
若是想安裝CPU版本,使用如下命令:
conda install tensorflow
若是在安裝過程當中出現錯誤,通常是網絡問題,你們懂的,使用ke-xue衝浪get

4.安裝完成後測試

首先輸入python進入python交互環境,而後輸入如下代碼:it

import tensorflow as tf 
print(tf.__version__)
print(tf.test.is_gpu_available())

預期結果應該以下圖,主要是最後輸出True,表示使用了GPU,這樣就表示安裝成功了。
QQ截圖20200717190332.pngio

相關文章
相關標籤/搜索