1.檢查顯卡驅動的版本 我的電腦是GTX 960M 對應的是CUDA8.0版本

https://developer.nvidia.com/cuda-toolkit-archive

進入該網址進行下載

2.CUDA8.0 對應 cudnn6.0的版本

https://developer.nvidia.com/rdp/cudnn-archive

3.tensorflow 應該是對應1.4.0 或者 1.3.0 版本(注意此時的python應該為3.6)

win+R 輸入cmd

pip install tensorflow-gpu==1.4.0

4.安裝之后完成測試

win+R 輸入cmd

輸入python 進入python環境

import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))

總結

以上是生活随笔為你收集整理的win10+NVIDIA GTX 960M+CUDA 8.0+cudnn6.0+tensorflow安装的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。