失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > 在anaconda中安装tensorflow-GPU版本

在anaconda中安装tensorflow-GPU版本

时间:2019-09-03 21:07:40

相关推荐

在anaconda中安装tensorflow-GPU版本

在anaconda中安装tensorflow-GPU版本

第一步:去anaconda官网下载,我下载的是anaconda (python3.7版本)

anaconda下载链接:点击link

进去之后网页最下面有要下载的版本

安装的具体步骤:点击

link

第二步:安装tensorflow的具体步骤

1、 anaconda安装之后进入Anaconda Prompt(anaconda)

2、进去之后创建一个虚拟环境

代码如下

conda create -n TF2.1 python=3.7

输入之后点击回车

用coda activate TF2.1

激活之后不要急着安装,因为下载会很慢,安装中科大镜像:具体步骤点击link,安装成功之后再进行以下安装操作

激活这个虚拟环境,进入虚拟环境之后安装

conda install cudatoolkit=10.1

安装过程中出现Proceed([y]/n)?都选y

安装成功之后:安装下面这个,步骤同上

conda install cudnn=7.6

成功之后安装

pip install tensorflow==2.1(不用指定GPU版本,默认CPU版本,GPU版本都会下载,GPU不能用的时候转为使用CPU)

安装之后测试一下,还是当前页面

输入

python回车

import tensorflow as tf 回车

tf.__version__回车

安装GPU版本需要Microsoft Visual C++ - Redistributabled的支持,木有安装的点击link自行安装,里面有链接

link

下载完之后如何在anaconda中使用刚才创建的虚拟环境

进入anaconda Prompt,激活刚才的虚拟环境

conda activate TF2.1回车,进入环境之后输入conda install nb_conda

回车

出现Proceed([y]/n)?都选y

安装之后,会出现jupyter notebook(TF2.1),点击进去

点击new,选择刚才创建的虚拟环境Python [conda env:TF2.1]

不需要时可以删除创建的虚拟环境

conda remove -n 虚拟环境名 --all

如果觉得《在anaconda中安装tensorflow-GPU版本》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。