問題描述
當我們使用tensorflow最新版本的時候https://github.com/tensorflow/tensorflow/r...,
預設的tensorflow pip包包含TF的CPU和GPU版本。 在以前的TF版本中,未找到CUDA庫會發出錯誤並引發異常,而現在庫會從動力學上搜尋正確的CUDA版本,如果找不到,則會發出警告(開頭的W代表 對於警告,錯誤具有E(對於致命錯誤則為F),並退回到僅CPU模式。實際上,警告之後,錯誤訊息也作為資訊訊息寫在日誌中(請注意,如果警告級別更高, 最低日誌級別(預設),您可能看不到資訊訊息。
詳細報錯:
2020-02-07 20:58:37.344417: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
2020-02-07 20:58:37.350291: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
問題是在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.x\bin\
中無法查詢到cudart64_101.dl
檔案。
至今在cuda 10.1版本中我們是可以看到這個檔案的,當我們下載新的cuda時,可能會報錯。
問題解決方法
- 下載visual studio :https://visualstudio.microsoft.com/zh-hans...
- 下載cuda 10.1版本:https://developer.nvidia.com/cuda-10.1-dow...記住哦要下載local版本的。大概有2g左右
- 下載cuda 10.2版本,https://developer.nvidia.com/cuda-download...
- 先安裝10.1版本,再將
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin\cudart32_101.dll
檔案複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin\
中,即可使用
- 先安裝10.1版本,再將
假如先下載10.2版本
先下載10.2版本後,後下10.1版本時不會產生新的dll檔案,我們需要將C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\
資料夾暫時移除,或者生成zip包並刪除資料夾再來重新下載10.1版本,就可獲得C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin\cudart32_101.dll
檔案。再加壓或者重新復原10.2資料夾後,再將C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin\cudart32_101.dll
檔案移入,即可。
本作品採用《CC 協議》,轉載必須註明作者和本文連結
文章!!首發於我的部落格Stray_Camel(^U^)ノ~YO。