[tensorflow-gpu][linux][cuda][cudnn][非root] 安裝

  1. 要裝tensorflow-gpu 先刪除cpu版本的tensorflow。
  2. 版本統(tǒng)一：
  • python版本3.6.6
  • cuda toolkit 9.0
  • cudnn 7
  1. 可參考 https://www.cnblogs.com/Climbing-Snail/p/9361621.html
  2. 其中上述鏈接的第三步，cudnn 下載需要NVIDIA注冊，選擇cuDNN library，
    下載文件為 cudnn-9.0-linux-x64-v7.3.1.20.solitairetheme8
cp cudnn-9.0-linux-x64-v7.3.1.20.solitairetheme8 cudann-9.0-linux-x64-v7.3.1.20.tgz
tar -xvf cudann-9.0-linux-x64-v7.3.1.20.tgz
image.png
  1. 別的步驟都可以參考以上鏈接來
  2. 我代碼里用到了CuDNNLSTM (會比LSTM快很多)，有的版本不能用，說一種我實驗成功的搭配：python 3.6.6, tensorflow-gpu 1.4.1, cuda 8.0, cudnn 6
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容