win10在本地安装tensorflow-gpu的详细记录

W three

已于 2022-09-30 17:04:55 修改

阅读量1.1k

点赞数 1

文章标签： tensorflow python 深度学习

于 2022-09-09 22:45:08 首次发布

本文链接：https://blog.csdn.net/weixin_43788752/article/details/126788034

版权

写在前面

刚接触机器学习，在本地配环境的时候有一步是需要安装tensorflow，但安装tensorflow, cuda, cudnn的版本一定是要对应的！不然就无法使用，装了好几遍才成功，记录一下安装过程。（PS.现在应该比较少这种情况：有比较老的显卡不支持GPU，那么在装tensorflow时只能安装CPU版本，就不用装cuda之类的，不要看这个了）

1.确定版本

先要查看电脑显卡支持的版本，这决定了安装的cuda版本，查看电脑显卡版本比较简单，可以自己去搜一下或者直接看这：

如何查看cuda的版本-百度经验 (baidu.com)

我的电脑比较老了，可以看到支持cuda8.0，那么我需要安装cuda8.0。根据cuda版本确定对应的tensorflow，cudnn版本，创建对应的python版本虚拟环境，看下面这个链接来确定版本：

https://tensorflow.google.cn/install/source_windows#gpu

找到符合的版本，我选择tensorflow_gpu-1.4.0，python3.6，cuDNN6.0，cuda8.0。下面开始安装。

2.安装

2.1 创建虚拟环境

安装anoconda，这个就不赘述了，打开Anoconda Prompt，创建一个新的虚拟环境用于安装，我创建了一个基于python3.6的环境，python36是我的环境名

conda create -n python36 python=3.6 #创建python版本为3.6的虚拟环境
conda activate python36 #创建完成以后激活虚拟环境

2.2 安装cuda，cuDNN，tensorflow

先添加清华源，安装前面看好的版本的cuda和cudnn，一定要在后面指定版本。

#添加清华源
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 --set show_channel_urls yes

#删除清华源改回默认源
#conda config --remove-key channels  #（记一下，怕忘了）

conda install cudatoolkit==8.0 #安装对应版本的cuda
conda install cudnn==6.0 #安装对应版本的cudnn

然后安装tensorflow

一般的版本通过添加清华源后都可以直接通过pip install 安装，你可以用conda search tensorflow --info 查看可以安装的版本。

conda search tensorflow --info #查看可以安装的版本，有就直接安装
#有些早期版本需要安装的依赖包，可以看上面显示的dependencies部分
pip install tensorflow-gpu=={版本}

但我需要安装的是tensorflow1.4，会显示找不到。于是改用下载安装包到本地安装。从这里下载对应的文件（应该基本都有）：

Links for tensorflow-gpu

（注意对应需要的版本）我下载的是tensorflow_gpu-1.4.0-cp36-cp36m-win_amd64.whl

将目录切换到安装包所在的位置（cd +路径）

#安装tensorflow前我参考其他教程先装了numpy和h5py(这一步可忽略，tensorflow安装时会携带安装对应版本)
pip install numpy==1.16.0
pip install h5py==2.10.0
#安装
#如果切换目录时存储盘不同要先更换盘再cd
# E:
# cd E:\software_exe
cd E:\software_exe #这是我放置的路径
pip install tensorflow_gpu-1.4.0-cp36-cp36m-win_amd64.whl #安装

这个时候可以进入python环境 import tensorflow as tf 看看有没有成功装上

#仍然是在终端上
python #进入python
import tensorflow as tf
#没有奇怪的显示就是成功了！！！！
#如果失败了，就看看别的方法再试试QAQ
quit() #退出python
#conda remove -n {你的环境名} --all #可以删除整个环境重新开始

在我成功装上tensorflow后，我使用的算法还需要装上对应版本的keras（好像在tensorflow2之前没有集成keras，需要另装）。找到对应keras的版本：

【汇总】Tensorflow 和 Keras 版本对应关系 - 知乎 (zhihu.com)

我需要的是keras2.0.8