以下是我自己安装tensorflow时的过程,我装的GPU版本,过程踩了不少坑,所幸磕磕绊绊也算解决。
1.安装anaconda3,选择安装位置。
https://www.anaconda.com/download/
记得安装过程中选择添加环境变量。
2.安装cuda和cudnn,我装的是tensorflow-gpu 2.1.0版本,查阅tengsorflow官网对应cuda和cudnn版本如下,cudnn我装的7.5.0版本没有装图片对应的7.6版本,具体原因下面安装会提到
tengsorflow官网网址如下
https://tensorflow.google.cn/install/source_windows
下载cuda10.1,选择安装位置
CUDA Toolkit 10.1 original Archive | NVIDIA Developer
有三个10.1版本的cuda,我选择Feb2019这个版本,cudnn也要选择相同月份的版本否则pycharm会报错
下载cudnn7.5.0
https://developer.nvidia.com/rdp/cudnn-archive
cudnn版本和cuda版本月份要相同,cuda选择Feb2019版本所以cudnn也是
下载后解压
把上图三个文件夹复制到cuda安装文件夹下
cuda和cuddnn安装成功
3.以管理员的身份运行命令提示符
输入conda,出现以下信息则说明anaconda安装成功
输入activate激活anaconda环境
输入以下命令创建tengsorflow虚拟环境,
conda create -n tf20 python=3.7
tf20是文件名,创建后会在envs文件夹下生成该文件夹,python=3.7是Python版本
输入以下命令进入tf20文件夹
conda activate tf20
在tf20文件夹下输入以下命令安装TensorFlow,这里我是用的是豆瓣的镜像源,下载速度会快一些
pip install tensorflow_gpu=2.1.0 -i https://pypi.douban.com/simple --trusted-host pypipypi.douban.com
在tf20文件夹下输入conda list验证TensorFlow安装成功
打开pycharm配置TensorFlow环境,打开设置,选Python解释器, 点添加
点现有环境
选择python.exe文件,点击确定就会就会加载TensorFlow
点击确定,网上复制粘贴测试代码
运行代码,出现以下结果说明TensorFlow安装成功
到这里TensorFlow安装成功
在tf20文件夹下输入以下命令安装jupyter notebook编辑器
conda install jupyter notebook
安装完成之后输入jupyter notebook进入jupyter编辑器
按Ctrl+C退出