安裝配置深度學習環境 TensorFlow 小記

自己的筆記本安裝

今年在美國入手了一個筆記本 Lenovo Legion Y530 (國行好像叫Y7000p) ,就隨意嘗試一下在自己的 laptop 上安裝一下深度學習環境,順便幫女朋友做個作業。
具體安裝方法請參考這位老哥的博客 Win 7/10下Anaconda/Python /TensorFlow-GPU/VS Code安裝配置 - senshuran的博客 - CSDN博客,幾乎反正安裝都沒什麼好說的,安裝過程都是沒什麼難點的。
我的電腦配置 Windows 10 家庭版 1809,8代 i7 , 雙通道 16G 內存,Nvidia GTX 1050 Ti 4GB 顯卡。

Python環境準備

我的電腦是 Win10 家庭版,我之前爲了做學校的作業,所以已經在電腦上安裝過 VSCode 和 Anaconda 3,還在其中創建了 python 2.7 的虛擬環境。

查詢版本

事先看了一下好像 Windows 版本的 TensorFlow 都只支持 python 3.5 以上的版本,順便查詢一下對應的 CUDA 版本。

第一個坑

一開始因爲 Anaconda 3 默認的是 python 3.7,所以我選擇了支持它的 CUDA 10.1 + TensorFlow-gpu 1.13.1,結果裝完之後運行的時候瘋狂提示缺少什麼 DLL 支持。不管使用什麼方法都沒有用,我也早都裝過什麼網上說的 VS2015 C++ 那個什麼包,各種依賴環境應該都是有的,依然沒有用。

最後還是乖乖的降版本,選擇了 CUDA 10.1 + TensorFlow-gpu 1.13.1,果然在 import tensorflow 的時候通過了;然後我試了一下 hello world ,又出錯了。

第二個坑

選擇 CUDA 10.1 + TensorFlow-gpu 1.13.1 的時候報錯我現在有點忘了,大致是說什麼 h5py 和什麼另一個包產生了衝突,網上說什麼要重新安裝,把 h5py 和那個包都降版本,嘗試了一下版本根本沒辦法匹配,降版本無濟於事。

最終還是找到了一個同樣使用 1050 Ti 的老哥寫的安裝博客 win10+1050ti+CUDA9.0+cudnn9.0的tensorflow-gpu環境搭建 - 豬八戒吃胡蘿蔔 - CSDN博客 ,所以我也選擇接着重裝,用的離線包,在去看女朋友的路上在火車上裝的。最終成功的版本是 CUDA 9.0 + cuDNN 7.6.0 for CUDA 9.0 + TensorFlow-gpu 1.11.0 。當然,因爲這個 tf 的版本只支持 python 3.6 ,所以我重新創建了一個支持 python 3.6 的 Anaconda 虛擬環境:

conda create -n tensorflow python=3.6 -y

以上。暫時就這麼多。

實習的公司的電腦要保密也不能多說,等哪天想寫了再補。。。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章