自己目前做視頻識別需要tensorflow-gpu加速喉酌,走了很多坑重裝好幾次系統(tǒng)矾克,終于實(shí)現(xiàn)粥诫,現(xiàn)記錄如下:pycharm+tensorflow+1050TI+Python3.6=
目前網(wǎng)上復(fù)制嚴(yán)重,好多沒有實(shí)踐可行性野建。主要問題如下:使用Anconda情況下cuda不用另外安裝娜谊;cuda不需要額外配置确买,cudnn也不需要配置。GPU版本查看使用https://www.techpowerup.com/gpuz/
第一步 安裝anconda
下載anconda?Anconda下載網(wǎng)址
進(jìn)入cmd設(shè)置國內(nèi)鏡像
修改鏡像源
https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes
創(chuàng)建環(huán)境
conda create -n 虛擬環(huán)境名稱 python=3.7
激活環(huán)境
activate 虛擬環(huán)境名稱
退出環(huán)境 deactivate
刪除環(huán)境 conda env remove -n 虛擬環(huán)境名稱
依賴的三方包
conda install tensorflow
conda install tensorflow-gpu
第二步 安裝cuda cudnn
調(diào)用GPU纱皆,現(xiàn)在conda命令不太好用拇惋,pip好用,但關(guān)聯(lián)功能沒有conda省事抹剩。
使用conda命令安裝?
conda install cudatookit 會自動安裝cudnn
第三步 驗(yàn)證cudn,cudnn 蓉坎,GPU 是否可用
切換到python命令行:運(yùn)行以下命令
from tensorflow.python.client import device_lib as _device_lib
local_device_protos = _device_lib.list_local_devices()
第四步 安裝PyCharm 配置
File->Default Setting->Project Interpreter->conda Environment
在add Local中添加本地虛擬環(huán)境。
注意問題?
conda install tensorflow-gpu 會自動關(guān)聯(lián)cuda及cudnn 查看版本是否匹配
python 版本與tensorflow-gpu版本是否匹配