自己目前做視頻識別需要tensorflow-gpu加速，走了很多坑重裝好幾次系統(tǒng)，終于實現(xiàn)，現(xiàn)記錄如下：pycharm+tensorflow+1050TI+Python3.6=

目前網(wǎng)上復(fù)制嚴重，好多沒有實踐可行性。主要問題如下：使用Anconda情況下cuda不用另外安裝；cuda不需要額外配置，cudnn也不需要配置。GPU版本查看使用https://www.techpowerup.com/gpuz/

第一步 安裝anconda

下載anconda?Anconda下載網(wǎng)址

進入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

第三步 驗證cudn，cudnn ，GPU 是否可用

切換到python命令行：運行以下命令

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版本是否匹配