ubuntu18.04安装tensorflow-gpu

转载：https://blog.csdn.net/u014670893/article/details/82467384

ubuntu18.04 安装tensorflow-gpu

2018年09月06日 20:35:47 折木 阅读数 6322

目录

驱动

CUDA Toolkit 9.0

CUDNN 7.0

libcupti

配置

安装tensorflow-gpu

测试

---

主要参考：https://medium.com/@taylordenouden/installing-tensorflow-gpu-on-ubuntu-18-04-89a142325138

驱动

https://blog.csdn.net/tonydz0523/article/details/83109343 

安装完成后最好重启下。

检查驱动是否安装成功

CUDA Toolkit 9.0

目前tensorflow只支持CUDA Toolkit 9.0。

下载地址：

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal

目前没有支持ubuntu18.04的CUDA Toolkit 9.0。选择17.10的版本，安装base installer一般够用了。

   
   
   
   

    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       sudo chmod +x cuda_9.
       
       
       
       0.176_384.81_linux.run
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       ./cuda_9.
       
       
       
       0.176_384.81_linux.run --
       
       
       
       override
      
      
      
      
     
     
     
     
   
   
   
   
   
   
   
   

installing with an unsupported configuration？时选择yes

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81？为了稳定最好选择no

样例可以不安装。

CUDNN 7.0

地址：https://developer.nvidia.com/rdp/cudnn-archive

   
   
   
   

    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       # 解压
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       tar -zxvf cudnn-
       
       
       
       9.0-linux-x64-v7.tgz
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      
 
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       # 复制相应文件
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       sudo cp -P cuda/lib64/libcudnn* 
       
       
       
       /usr/local
       
       
       
       /cuda-9.0/lib64/
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       sudo cp  cuda/
       
       
       
       include/cudnn.h /usr/local/cuda-
       
       
       
       9.0/
       
       
       
       include/
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      
 
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       # 所有用户可读
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       sudo chmod a+r /usr/local/cuda-
       
       
       
       9.0/
       
       
       
       include/cudnn.h /usr/local/cuda/lib64/libcudnn*
      
      
      
      
     
     
     
     
   
   
   
   
   
   
   
   

libcupti

sudo apt-get install libcupti-dev
   
   
   
   

配置

在~/.bashrc中加入

   
   
   
   

    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       export PATH=
       
       
       
       /usr/local
       
       
       
       /cuda-9.0/bin${
       
       
       
       PATH:+
       
       
       
       :${PATH}}
      
      
      
      
     
     
     
     
    
    
    
    

     
     
     
     

      
      
      
      

     
     
     
     
     
     
     
     

      
      
      
      

       
       
       
       export LD_LIBRARY_PATH=
       
       
       
       /usr/local
       
       
       
       /cuda/lib64
       
       
       
       :${
       
       
       
       LD_LIBRARY_PATH:+
       
       
       
       :${LD_LIBRARY_PATH}}
      
      
      
      
     
     
     
     
   
   
   
   
   
   
   
   

安装tensorflow-gpu

根据需要进行修改

pip install --upgrade tensorflow-gpu
   
   
   
   

测试

成功!