Ubuntu18.04安装Tensorflow+cuda+cuDNN

本文写的比较简单,期间遇到的一些小麻烦,本身不认为成为阻碍,因此没有详细写。
若有疑问能够联系QQ:2922530320python

Pycharm

Pycharm使用Anaconda

Pycharm 在新建项目的时候能够指定Conda环境,前提是已经安装了Conda。linux

若是建立项目的时候没有指定conda环境:git

Setting => Project => Project Interpreter 里面修改 Project Interpreter , 点击齿轮标志再点击Add Local为你某个环境的python.exe解释器就好了github

Anaconda

Anaconda 安装包能够到 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 下载。shell

若是pip安装软件包速度很慢,可使用清华的镜像库ubuntu

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

介绍:Anaconda会建立一个独立的python环境,针对某个应用下载其须要的依赖包,并保存在一个自定义的目录(目录名经过Anaconda建立项目时指定),以后咱们在Anacoda的环境中编写代码,代码引用的包就是Ananconda为此项目独立下载的包,不会影响到宿主机和其余环境。python2.7

好比你要在learn环境中编写程序, 那么就修改成~/anaconda3/envs/learn, 能够看到这时候下面的依赖包也变成了learn环境中的包了.接下来咱们就能够在pycharm中愉快的编码了.编码

经常使用的命令

查看全部的环境

conda env list

安装第三方包

conda install requests
或者 pip install requests

卸载第三方包

conda remove requests
或者 pip uninstall requests

查看环境包信息

conda list

导入导出环境

conda env export > env.yaml
conda env create -f env.yaml

切换环境

activate python35          #Windows
source activate python35   #linux&Mac

建立环境

conda create -n <env_name> python=3.6
建立一个环境,指定环境名称,并指定使用的python版本

删除环境

conda remove -n <env_name> --all
删除环境以及下属的全部包

安装Tensorflow

一、正常流程大概是第一步这样的,可是,咱们通常不走寻常路。url

tensorflow目前为止支持到cuda9.0,而ubuntu18.04上默认的cuda是9.1,安装方式很简单:sudo apt install nvidia-cuda-dev。本人懒的重装cuda,由于还要降级gcc到6.0。既然官方不支持,就找非官方的。因此跳过第一步(其实第一步也不完整)。code

在anaconda环境下执行

pip install tensorflow-gpu

cuda历史版本下载

https://developer.nvidia.com/cuda-toolkit-archive

具体执行命令和安装方式在下载的地方已经有说明了。

二、网上大神已经编译好的tensorflow,目前已经更新到1.8了。

https://github.com/mind/wheels/releases/

有三个版本,cp27,cp35和cp36,分别对应的是python2.7,3.5和3.6。

下载[TensorFlow 1.8 (GPU, CUDA 9.1, cuDNN 7.1, no MKL)]这个标题下面的包。

在本身的anaconda环境安装下载好的tensorflow

pip3 install 上面下载的 .whl 文件

cuDNN安装

官网下载须要注册,网上有人提供了下载好的。

连接: https://pan.baidu.com/s/1DiQYF8KFnuwfVDnmlxBw7w 密码: ga6s

tar -xf cudnn-9.1-linux-x64-v7.1.tgz
cd cuda
sudo cp include/cudnn.h /usr/include/
sudo cp lib64/* /lib/

最后一步

重启系统

相关文章
相关标签/搜索