windows10+anaconda3+python3.7+pycharm+gpu–tensorflow 2.1
NVIDIA GeForce GTX 1650版本441.22安裝的CUDA版本是10.2.9
CUDA
有關顯示卡,顯示卡驅動,cuda ,cudnn是什麼,以及之間的關系可自行搜尋
在左下角搜尋框搜尋NVIDIA進入控制台,查詢目前版本
算力需大于3.1
查詢CUDA版本和顯示卡驅動版本 需按對應下載下傳
CUDA下載下傳 下載下傳非常慢 😢,安裝就是一直無腦下一步
測試win+R: nvcc -V
出現上圖證明成功。
cuDNN
cuDNN下載下傳 直接下載下傳要注冊,右鍵選擇複制連結,在打開迅雷即可,或者直接右鍵迅雷下載下傳,我選擇的是for cuda9.0的版本,沒選擇最新的,好像問題不大,也有可能比較慢,
最後得到一個zip檔案,解壓後得到三個檔案夾,如下所示:
将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
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連結
再次運作成功
希望對你有所幫助!