天天看點

最新tensorflow安裝

windows10+anaconda3+python3.7+pycharm+gpu–tensorflow 2.1

NVIDIA GeForce GTX 1650版本441.22安裝的CUDA版本是10.2.9

CUDA

有關顯示卡,顯示卡驅動,cuda ,cudnn是什麼,以及之間的關系可自行搜尋

在左下角搜尋框搜尋NVIDIA進入控制台,查詢目前版本

最新tensorflow安裝

算力需大于3.1

查詢CUDA版本和顯示卡驅動版本 需按對應下載下傳

CUDA下載下傳 下載下傳非常慢 😢,安裝就是一直無腦下一步

測試win+R: nvcc -V

最新tensorflow安裝

出現上圖證明成功。

cuDNN

cuDNN下載下傳 直接下載下傳要注冊,右鍵選擇複制連結,在打開迅雷即可,或者直接右鍵迅雷下載下傳,我選擇的是for cuda9.0的版本,沒選擇最新的,好像問題不大,也有可能比較慢,

最後得到一個zip檔案,解壓後得到三個檔案夾,如下所示:

最新tensorflow安裝

将bin ,include 和lib檔案夾的内容分别複制到cuda安裝路徑下與cuda的bin ,include 和lib檔案夾合并。

測試安裝是否成功

win+R :cmd

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite

.\deviceQuery.exe顯示如下結果說明安裝成功:

最新tensorflow安裝
Tensorflow

win+R

python -m pip install --upgrade pip
pip install tensorflow-gpu==2.1
           

首先更新pip,第二句容易下載下傳逾時,不行則換成下面這句

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow 
           

指定鏡像源會比較快,安裝需要一點時間

pycharm

pycharm安裝

建立一個項目檔案,将anaconda和pycharm綁定

file->settings->project interpreter ->右邊小齒輪->show all ->+号->一般在c盤user路徑下選擇anaconda檔案中的Python.exe

建立一個python程式

import tensorflow as tf
tf.__version__
           

首次運作出現如下錯誤

Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found

Ignore above cudart dlerror if you do not have a GPU set up on your machine.

通過這篇部落格順利解決

先來排查錯誤,顯示“cudart64_101.dll not found”,那是因為我安裝的cuda為10.0版本的,“cudart64_101.dll”是cuda10.1這個版本中才含有的(cuda10.2貌似也沒有)。

是以我們需要手動将“cudart64_101.dll”添加到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin”中去。

另附cudart64_101.dll連結

再次運作成功

最新tensorflow安裝

希望對你有所幫助!

繼續閱讀