這里面有很多坑,最大的坑是發現各方面都裝好了結果報錯 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模型了。