安装时间:2019年12月,系统:win10的64位,显卡:Nvidia GTX1050 Ti(4G)
一、下载及安装Anaconda
这里选择Python3.7、64位下载,Tensorflow2.0不兼容Python2.7。
官网:https://www.anaconda.com,清华镜像:https://mirror.tuna.tsinghua.edu.cn/anaconda/archive/
二、NVIDIA显卡驱动安装/更新
确保NVIDIA显卡的计算能力应该大于3.0,我的是1050Ti,算力应该是大于6.1的。
算力查看:https://developer.nvidia.com/cuda-gpus
NVIDIA驱动最低版本41.0x,可通过NVIDIA控制面板查看,官网下载更新:https://www.nvidia.cn,安装后重启电脑。
三、Tensorflow2.0-GPU安装
安装好Anaconda及NVIDIA驱动后,找到“Anaconda Prompt(Anaconda3)“打开,输入”conda -V“,可查看当前安装Anaconda的版本。
如果不安装在base中可通过参考下列指令,新建一个tensorflow-gpu的环境,方便后面管理。
conda create -n XXX --clone root 创建XXX环境并且复制base的环境
conda create -n XXX python=3.8 创建XXX环境,python版本3.8
conda info -e 查看现有的环境与当前所属环境
conda activate XXX 切换当前运行环境,xxx为环境名称
conda deactivate 切换回初始(base)环境
conda remove --name xxx --all 删除叫XXX的环境
输入”conda install tensorflow-gpu==2.0“,开始下载安装。
如果出现下载失败可考虑添加国内源,以清华源为例。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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 --set show_channel_urls yes
四、打印Hello TensorFlow
五、Anaconda与Pycharm关系
anaconda是一个开源的Python发现版本,包含了大量库的python包,生成的文件后缀为.ipynb,原始python后缀为.py。 pycharm是IDE编辑器,pycharm可以用原始python也可以用anoconda的python。
六、镜像办法
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.0.0a0
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-cpu==2.1.0rc0
注意以上格式为:pip install -i 镜像网址 你要的tensorflow版本号
镜像网址包括: