在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++ 2015-2019 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