tensorflow是google推出的深度學習開源框架。相比於keras、caffe等框架,tensorflow在github上的star數量、fork數量、contributor數量都一騎絕塵。,tensorflow官方已經宣布原生支援windows作業系統,但目前,仍主要通過cuda支援nvidia的gpu,amd系顯示卡可通過opencl支援,但成熟度較低。
本人的安裝環境為:wiindows 10, anaconda python36, pycharm
1.由於keras需要以tensorflow作為後端,所以在使用gpu版本的keras之前,需要安裝tensorflow的gpu版本。
2.關於tensorflow-gpu安裝,請參考本人的這篇部落格:
3.在成功完成第三步的基礎上,再在anaconda的prompt下輸入 pip install keras,它會自動附帶安裝所需的其他庫和theano,需要注意的是,此時keras會自動選用tensorflow作為其後端,而tensorflow則會在有可用gpu時自動呼叫gpu執行。
4.安裝成功:
4. 開啟pycharm, 在python console下輸入import keras
會顯示using keras backend,安裝成功。
CUDA tensorflow的GPU版本安裝
驗證安裝成功 注意cuda的版本,如果遇到了could not load dynamic library cudart64 100.dll dlerror cudart64 100.dll not found,說明cuda版本不對。cudart64 100.dll 中的100說明需要的是10.0版本...
gpu安裝記錄
系統環境 centos 7.6,無桌面版 一 安裝驅動 a 禁用 nouveau 只有在禁用掉 nouveau 後才能順利安裝 nvidia 顯示卡驅動,禁用方法就是在 sudo vim etc modprobe.d blacklist.conf檔案中新增一條禁用命令 可參考 與 centos部分 ...
GPU版pytorch的環境安裝記錄
1.安裝gpu版本pytorch教程 避坑 2.torch的cuda版本選擇是10.2,電腦上cuda版本可以高於torch的版本,我的是cuda11。win10安裝cuda和cudnn的正確姿勢 3.cuda版本的 改寫。pytorch學習系列 十六 如何使用cuda進行訓練?gpu 上的 pyt...