1. 程式人生 > 實用技巧 >【安裝教程】win10+Python3.6+TensorFlow-GPU,那些筆者踩過的坑!直接pip安裝TensorFlow

【安裝教程】win10+Python3.6+TensorFlow-GPU,那些筆者踩過的坑!直接pip安裝TensorFlow

新手必看的TensorFlow安裝教程,話不多說直接上乾貨。

先說一下博主的電腦配置,Dell,i5,win10。不過這些都不重要,但是電腦一定要帶NVIDIA獨顯

**

一、先去看一下自己是否安裝了Visio Studio,沒有的話,先安裝,安裝了直接下一步**

https://blog.csdn.net/xucongyoushan/article/details/86017602
按照上面那個連結操作即可。

二、tensorflow下載+安裝+對應版本!

先去1下載對應的版本並安裝,如果安裝失敗了,看2怎麼安裝的,看了2之後有疑問的地方看3,在3裡面有一個解決方案:在路徑 D:\ProgramFiles\Microsoft Visual Studio 14.0\Common7\IDE下,開啟cmd(管理員身份): “devenv /setup” 在指定目錄下開啟cmd

,怎麼設定(大神請繞路),請看4
1.https://blog.csdn.net/gdc6916/article/details/105384369
2.https://blog.csdn.net/zzpong/article/details/80282814
3.https://blog.csdn.net/jin739738709/article/details/80819441
4.https://blog.csdn.net/hiudawn/article/details/80701935

基本上博主參考的論文都在這裡了!】

如果還沒成功或者怎麼測試是否成功,請看補充說明。
補充說明:
1.在連結1裡面博主先下載了TensorFlow-GPU 1.6版本,但是測試的時候版本太低了,出現如下問題,

在這裡插入圖片描述
意思是說TensorFlow版本太低,所以博主,升級到了NVIDIA所支援的最高TensorFlow1.12,然後再去測試是成功的哈。
在這裡插入圖片描述
2.怎麼測試,安裝結束後,看看自己安裝成功了嘛。

對於CUDA的測試,開啟命令列(cmd),輸入nvcc -V
如果是下面這個圖片恭喜你CUDA安裝成功了!
如果是

對於TensorFlow-GPU的測試,
先在命令列,輸入Python
在這裡插入圖片描述
然後輸入下面三句話,有的人可能會有疑惑,直接輸入import tensorflow as tf 不可以嘛,筆者直接輸入了之後就報錯了,所以我猜想是需要先匯入,然後才可以改名字。
在這裡插入圖片描述
對於TensorFlow-GPU的測試,接著輸入tf.test.is_gpu_available()

在這裡插入圖片描述
如果看到true就說明你成功了,恭喜你。

  1. pip如果版本太低就不能下載TensorFlow,所以可以先升級pip,通過 python -m pip install --upgrade pip這條語句升級。

因為太不容易,所以寫下來,供大家參考。