Windows & Ubuntu Anaconda安装tensorflow-gpu2.0

本文拟安装tensorflow-gpu2.0,打算使用3.6版本的python,安装Anaconda3-5.2.0。

安装Anaconda

利用Anaconda可以方便地在同一台计算机上安装不同的环境。Anaconda安装包使用清华镜像下载速度更快。

Windows系统安装Anaconda

  1. win10从上述链接中下载Anaconda3-5.2.0-Windows-x86_64.exe(64位);
  2. 执行Anaconda安装文件,像往常安装软件一样安装即可,安装过程中注意勾选将Anaconda加入环境变量(Add Anaconda to my PATH environment variable);
  3. Win+R打开cmd,输入conda --version验证Anaconda是否安装成功。

Ubuntu系统安装Anaconda

  1. Ubuntu从上述链接中下载Anaconda3-5.2.0-Linux-x86_64.sh(64位);
  2. 打开终端进入相应的文件夹,输入bash Anaconda3-5.2.0-Linux-x86_64.sh,之后一步一步安装即可,注意提示信息适时输入yes加入环境变量,另外在提示信息"Do you wish to proceed with the installation of Microsoft VSCode? [yes|no]"后,输入no;
  3. 打开终端,输入conda --version验证Anaconda是否安装成功。

确保显卡驱动版本合适

要在tensorflow中使用GPU需要借助于cuda、cudnn,而cuda的不同版本对于GPU的驱动版本有一定的要求,故需要确保显卡驱动版本符合要求。具体cuda版本对GPU驱动版本的要求可查看cuda官网相关网页的table2,如下图(红框框出的为我们需要的)。
Windows & Ubuntu Anaconda安装tensorflow-gpu2.0_第1张图片

Windows系统

进入NVIDIA控制面板,点击系统信息,即可查看当前驱动程序版本。
Windows & Ubuntu Anaconda安装tensorflow-gpu2.0_第2张图片图中显示的版本已经符合要求,如果你的GPU驱动版本低于cuda官网提供的表格中的要求,需要进入NVIDIA驱动程序下载,下载符合要求的GPU驱动,下图为我下载的选项,下载可执行文件后像安装软件一样安装即可。
注:Game Ready驱动程序(GRD)或者Studio 驱动程序(SD)均可,使用Game Ready驱动程序tensorflow-gpu也能正常使用。
Windows & Ubuntu Anaconda安装tensorflow-gpu2.0_第3张图片

Ubuntu系统

具体什么流程我记不清了,但需要先使nouveau停止,再安装相关的nvidia驱动,我有以下记录:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo ubuntu-drivers devices
sudo apt-get install nvidia-430 nvdia-prime nvidia-settings
lsmod | grep nouveau	#无输出说明nouveau停止
sudo apt-get --purge remove nvidia*	#卸载未安装成功的驱动
sudo apt autoremove
sudo reboot
nvidia-settings
uname -r sudo apt install linux-headers-$(uname -r) 确保内核与两相关配件版本相同

安装tensorflow-gpu2.0

  1. 打开Windows的cmd、Ubuntu的终端创建相应的环境,此处命名为"tf2",安装过程需要一定的时间;
conda create -n tf2 python=3.6
  1. 激活环境;
conda activate tf2
  1. 安装tensorflow-gpu2.0,安装过程需要较长的时间。
pip install tensorflow-gpu==2.0.0
conda install cudnn cudatoolkit=10.0 numba

验证安装结果

打开windows的Anaconda Prompt或Ubuntu的终端,依次输入:

conda activate tf2
python3
import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()

查看输出的提示判断功能是否正常。

安装方便的编辑器jupyter notebook

conda activate tf2
pip install jupyter notebook

需要使用jupyter notebook的时候进行以下操作,之后会在浏览器中出现操作界面:

conda activate tf2
#确保Anaconda Prompt或者终端进入你需要编辑的文件的文件路径
jupyter notebook

pip和conda加速安装的tips

使用pip安装遇到"socket.timeout:The read operation timed out"的问题

pip --default-timeout=100 install -U xxxxxx

pip用清华源进行安装:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple xxxxxx

设置conda用清华源:

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

注:以上xxxxxx代表需要安装的库。

你可能感兴趣的:(Windows & Ubuntu Anaconda安装tensorflow-gpu2.0)