GPU版本的Keas安裝

2021-08-30 04:33:36 字數 758 閱讀 7254

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...