2017年9月2日 星期六

win10安裝TensorFlow-GPU

到Python 官網下載並安裝 python 3.5

pip3 install --upgrade tensorflow-gpu

更新nvidia driver

下載cuda toolkit,在安裝時僅選擇安裝cuda選項

下載cudnn 5.1 for cuda8.0與cudnn 6 for cuda8.0,分別解壓縮至.../cuda/cudnn_v5, .../cuda/cudnn_v6資料夾放入C槽,將.../cuda/cudnn_v5/bin, .../cuda/cudnn_v6/bin分別加入至環境變數

測試:
python
import tensorflow as tf
hello = tf.constant('Hello TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

沒有留言:

張貼留言