Ubuntu gpu tensorflow环境搭建

安装cuda

1. 访问https://developer.nvidia.com/cuda-80-ga2-download-archive，下载相应的安装包（这里安装cuda 8.0,sudo apt-get install cuda-8-0，目前tensorflow 1.4 不支持9.1,注意，后面cudnn对应的版本也要改）

 

2. 放到服务器上面，执行一下代码：

 

3. 命令行测试是否安装成功：

 

4. 设置环境变量：

  

5. （卸载专用命令，不卸载就不要执行）

   

 

 

安装 cuDNN

1. 访问https://developer.nvidia.com/rdp/cudnn-download，下载安装包

   

2. 执行命令，安装

 

3. 验证安装成功

  

4. 然后执行pip install tensorflow-gpu安装成功就行好了。