要配置Tensorflow的GPU版本,需要先配置好CUDA和cuDNN。由于我要配置Tensorflow的1.4.0版本,所以选择的是CUDA8.0和cuDNN
v6.0版本。具体需要怎么选择可以参考下图(图片来源于博客:https://blog.csdn.net/weixin_39704651/article/details/79605585):



关于CUDA8.0和cuDNN v6.0的配置可以参考我前面的博客:

* Ubuntu16.04深度学习环境配置——NVIDIA显卡驱动、CUDA8.0、cuDNN v6.0
<https://blog.csdn.net/allenirving/article/details/80401117>
在进入本文的步骤之前请先确保CUDA和cuDNN的正确安装。

1、Anaconda的安装

参考博客:https://blog.csdn.net/Super_jm_/article/details/78397074

这里我选择的是基于Python3.5的Anaconda3,根据官网的描述,4.2.0版本的anaconda是内置Python3.5的。到这个网页
<https://repo.continuum.io/archive/>寻找对应的版本,即Anaconda3-4.2.0-Linux-x86_64.sh文件。

下载完成后,进入到下载保存目录,执行如下命令进行安装:
bash Anaconda3-4.2.0-Linux-x86_64.sh
一路yes下来就行了。安装完后记得查看~/.bashrc
文件中是否添加了anaconda的路径,如果是一路yes下来的话是自动添加了的。如果没有的话,需要手动添加:
export PATH="/anaconda3的安装路径/bin:$PATH"
记得根据自己的实际情况,填写anaconda3的安装路径。之后source一下使其生效:
source ~/.bashrc
在终端输入python之后,如果输出如下图,则安装成功。



2、更换pip源、conda源、apt-get源(可选操作)


换源可以加快命令使用时的下载速度,但可能会带来不必要的麻烦,可以自己权衡利弊做决定。

1)更换pip源
cd sudo mkdir .pip sudo vi ~/.pip/pip.conf
在新建的pip.conf文件中添加如下内容:
[global] index-url = https://pypi.douban.com/simple
然后按下esc键,输入:x或者:wq,回车保存并退出,这样即可将pip源更换为国内豆瓣镜像。

2)更换conda源
sudo vi ~/.condarc
输入以下内容:
channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ -
defaults show_channel_urls: true
然后按下esc键,输入:x或者:wq,回车保存并退出。

3)更换apt-get源

这里我把apt-get源换成了中科大的镜像。


首先备份旧的文件:
sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak
然后修改文件:
sudo vim /etc/apt/sources.list
在文件的开头或者结尾处添加以下内容:
deb http://mirrors.ustc.edu.cn/ubuntu/ xenial main restricted universe
multiverse deb http://mirrors.ustc.edu.cn/ubuntu/ xenial-security main
restricted universe multiverse deb http://mirrors.ustc.edu.cn/ubuntu/
xenial-updates main restricted universe multiverse deb
http://mirrors.ustc.edu.cn/ubuntu/ xenial-proposed main restricted universe
multiverse deb http://mirrors.ustc.edu.cn/ubuntu/ xenial-backports main
restricted universe multiverse deb-src http://mirrors.ustc.edu.cn/ubuntu/
xenial main restricted universe multiverse deb-src
http://mirrors.ustc.edu.cn/ubuntu/ xenial-security main restricted universe
multiverse deb-src http://mirrors.ustc.edu.cn/ubuntu/ xenial-updates main
restricted universe multiverse deb-src http://mirrors.ustc.edu.cn/ubuntu/
xenial-proposed main restricted universe multiverse deb-src
http://mirrors.ustc.edu.cn/ubuntu/ xenial-backports main restricted universe
multiverse保存退出即可。

3、通过Anaconda安装Tensorflow


首先需要新建一个conda环境,命名为tensorflow-gpu:
conda create -n tensorflow-gpu python=3.5
要注意Python版本需要写对。


待创建完成之后可以使用以下命令管理这个环境:
source activate tensorflow-gpu #激活环境 source deactivate tensorflow-gpu #关闭环境
在激活环境的情况下输入以下命令进行安装tensorflow的gpu版本:
pip install tensorflow-gpu==1.4.0
1.4.0即为版本号,可以自己更改。安装tensorflow的CPU版本的话就把上面的tensorflow-gpu换成tensorflow。

之后等候安装即可。安装完后会提示pip的版本有点低,需要更新,安装它给的提示命令进行更新即可。

接着,在环境激活的情况下,输入python,然后输入
import tensorflow as tf
如果不报错的话,说明安装成功。

也可以试一下下面这个稍微完整一点的代码:
import tensorflow as tf hello = tf.constant(‘hello tensorflow!’) sess =
tf.Session() print(sess.run(hello))
用完之后记得要关闭环境!!!


4、PyCharm的安装及配置

1)PyCharm的安装

首先去官网下载PyCharm的community版本。

下载完成后,解压到想要安装的目录,然后执行以下命令进行运行:
# 假设已进入pycharm的解压目录 cd bin bash pycharm.sh
然后就是一些基础设置了,完成之后可以看到以下界面:



为了以后使用方便,可以将pycharm的路径添加到~/.bashrc文件中:
vim ~/.bashrc
在文件末尾添加:
export PATH=/pycharm的解压路径/bin:$PATH
记得更改好路径,保存退出,之后再source一下使其生效。之后就可以直接在终端输入pycharm.sh即可启动PyCharm了。

2)PyCharm的配置

打开PyCharm之后,点击右下角的Configure,选择Settings,在左侧选择Project Interpreter,然后点击右侧面板的小齿轮,选择
Add...,在弹出的窗口左侧选择Conda Environment,接着选择Existing environment
,这时候它会自动识别出刚刚创建好的conda环境,然后勾选Make available to all projects,点击OK即可。

创建新项目的时候注意选择好Project Interpreter,因为刚刚的设置可能会不能应用到新项目,需要手动选择一下。

接下来就可以开始学习Tensorflow了!

友情链接
KaDraw流程图
API参考文档
OK工具箱
云服务器优惠
阿里云优惠券
腾讯云优惠券
华为云优惠券
站点信息
问题反馈
邮箱:[email protected]
QQ群:637538335
关注微信