本文描述的是如何在windows 10 64位操作系統(tǒng)安裝TensorFlow，采用預(yù)編譯版本。

前提條件

一塊支持CUDA 8.0的nvidia顯卡。

可以在官方網(wǎng)站查詢顯卡的兼容性信息。

安裝nvidia cuda 8.0

https://developer.nvidia.com/cuda-downloads

從官方網(wǎng)站下載安裝包，一路next就可以了。

確保cuda的路徑信息加入PATH，不然運(yùn)行會找不到nvcuda.dll。

安裝cuDNN

https://developer.nvidia.com/cudnn

下載安裝包后解壓縮，并把cudnn64_5.dll的路徑加入PATH。

安裝Anaconda

從https://www.continuum.io/downloads下載軟件并安裝。

安裝TensorFlow

conda create --name tensorflow python=3.5

activate tensorflow

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl

驗(yàn)證安裝

#coding=utf-8 #單機(jī)單卡 #對于單機(jī)單卡，可以把參數(shù)和計(jì)算都定義在gpu上，不過如果參數(shù)模型比較大，顯存不足等情況，就得放在cpu上 import tensorflow as tfwith tf.device('/cpu:0'):#也可以放在gpu上w=tf.get_variable('w',(2,2),tf.float32,initializer=tf.constant_initializer(2))b=tf.get_variable('b',(2,2),tf.float32,initializer=tf.constant_initializer(5))with tf.device('/gpu:0'):addwb=w+bmutwb=w*bini=tf.initialize_all_variables() with tf.Session() as sess:sess.run(ini)np1,np2=sess.run([addwb,mutwb])print(np1)print(np2)

--結(jié)束--

超強(qiáng)干貨來襲 云風(fēng)專訪：近40年碼齡，通宵達(dá)旦的技術(shù)人生

總結(jié)

以上是生活随笔為你收集整理的Windows10安装TensorFlow GPU版本的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。