win10 + 獨顯 + Anaconda3 + tensorflow_gpu1.13 安裝教程(跑bert模型)


這里面有很多坑,最大的坑是發現各方面都裝好了結果報錯  Loaded runtime CuDNN library: 7.3.1 but source was compiled with: 7.4.1,這是由於最新的tensorflow1.13需要用

Cudnn7.4.1編譯。這個問題,StackOverflow上有人問到,但是目前依然未有人給出解決辦法,下文會詳述。

1. 去nvida官網下載顯卡對應的驅動

2.下載安裝Anaconda3,安裝時勾選加入環境變量

3. 進入Anaconda Prompt 創建環境 conda create -n tensor1 python=3.6,然后進入環境 activate tensor1

4. 查找可用的 anaconda show anaconda/tensorflow-gpu 會給出 gpu版本的tensorflow下載地址,然后復制 conda install --channel https://conda.anaconda.org/anaconda tensorflow-gpu 到命令行,進行下載安裝(會安裝CudaTooltikit10.0 和 cudnn7.3.1)

5. 安裝后命令函行輸入python, 輸入import tensorflow as tf 不會報錯,這說明到目前為止安裝正常。但是我們在pycharm中嘗試import tensorflow as tf 就會報Loaded runtime CuDNN library: 7.3.1 but source was compiled with: 7.4.1錯誤。解決辦法如下。

6.請打開你的Anaconda安裝路徑(比如:我的路徑是C:\Users\78190\Anaconda3\pkgs),你會發現這有個

需要把它替換成cudnn-7.4.1-cuda10的對應版本,這里需要去Nvida官網去下載cudnn-7.4.1-cuda10,這里需要注冊用戶名才能下載,不要慫,慢慢來。

將下載好的下面的文件去覆蓋C:\Users\78190\Anaconda3\pkgs\cudnn-7.3.1-cuda10.0_0\Library下的對應文件夾。

7. 除此之外還需要替換C:\Users\78190\Anaconda3\envs\tensor1\Library下的三個文件

這三個文件夾里還有需要替換的東西,在這三個文件夾中分別找到:

 

從下載的cudnn7.4.1中找到對應的文件,去覆蓋這三個文件。

到此,即可成功的跑深度學習的bert模型了。

 

 

 

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM