闲话少说,先上anaconda必须掌握的几条语句:
创建一个新的环境,指定python 版本:
conda create -n XXX python=3.X
conda activate XXX
conda deactivate XXX
激活、关闭环境
conda info -e
查看所有环境
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
清华源有时候也会有问题,切换回默认源
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple keras==2.1.5
pip换源
屌话一堆: 来个图,注意一定要将所有的版本号对应起来,本人没有试过不对应的后果,有试过的hxd麻烦告诉我一下。
这里注意一下,用conda install 命令安装tf 的时候自动安装相应的cuda等包,可以一步到位,但是版本不是你自由选择的。
为了可以自由选择我们需要的版本,我们用conda install 安装cuda和cudnn;使用pip命令安装我们需要的tensorflow版本。
conda install cudatoolkit=10.1
conda install cudnn=7.6
我在另一台设备上配置的时候发现下载一半,总是报错,把源切换回默认源就好。
可能有细心的小伙伴发现,tf2.1-2.3版本对应的cuda 和 cudnn版本完全一致,而且大部分常见的工作是用不到特别高版本的tf的,所以安装一个支持广泛地版本是我们常见的选择。
完成上述步骤之后,就可以安装tensorflow-GPU了
pip install tensorflow-gpu==2.2
到这里,就安装好了tensorflow,用anaconda安装tensorflow无需去Nvidia官网上下载cuda和cudnn且管理非常方便。