之前一直用的win10版本的teensorflow,最近感觉不会Linux有点过时,索性开始搞起来。

搞了一天多,重装了4、5次系统,特记录下自己的路程。其实大部分是按照官网的顺序来的。

一、安装前的准备工作----安装puthon3,pip3,virtualenv:

python3 --version
pip3 --version
virtualenv --version

sudo apt update
sudo apt install python3-dev python3-pip
sudo pip3 install -U virtualenv  # system-wide install

二、安装显卡驱动:

看别人又是禁用这个又是禁用那个的,才发现自带的才是最好的。在software updates中选择驱动跟新安装即可。

三、安装cuda10.0:

这里要注意一下,cuda官网中支持ubumtu18.04的最低版本为cuda10.0,低版本也许兼容?我没试。本人一开始没注意下载了cuda10.1,结果目前tensorflow最高版本1.13.1才支持到10.0,搞了很久。

在官网找到相应的下载就好。

文件下相应的安装方法都下好了,直接照着搬上去。注意:第二行的命令version代表的是版本号,第一行执行完后也会提示执行的,别搞错了就好。安装很顺利,等就完了。

四、安装cudnn7.5

去官网找对应版本如图:

我下载的是deb版本,安装就一句话

sudo dpkg -i 文件名

至此,一个基础班的就搞好了,现在安装tensorflow

五、构建虚拟环境

virtualenv --system-site-packages -p python3 ./venv

激活虚拟环境

source ./venv/bin/activate  # sh, bash, ksh, or zsh

升级pip

pip install --upgrade pip

安装tensorflow

pip3 install tensorflow-gpu

测试

python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

没有出错的话就是安装好了。

 

很多都是参考官网的教程。有一句话说得好,官网的才是最好的,别人的说白了也只是别人自己特定情况下的罢了(liao)

 

就扯到这,玩tensorflow去了

Logo

更多推荐