ubuntu16.04 + cuda8.0安装

本文记得不够细致，有待细查

下载cuda.run，我自己的网盘也有

网上下载cuda.run

1.安装cuda

chmod +x ./sudo sh cuda_8.0.61_375.26_linux.run

sudo ./sudo sh cuda_8.0.61_375.26_linux.run

(1)运行后，先出来协议0%，按空格下翻，最后选择accept，

(2)如果 nvidia的驱动已经安装了，选择no。

(3)后面就yes 或者回车，直到安装成功

2.安装cudnn

2.安装cuDNN

下载cuDNN，从https://developer.nvidia.com/rdp/cudnn-download，注册，然后下载，

注意事项：

有可能由于最新cuDNN不稳定，导致后续caffe工程编译失败，报出如下错误，详情参考第三个链接，这时可以回退一下，下载较新版本的cuDNN；

拷贝cuDNN库文件到cuda目录下: 

  

   

    

     

    
    

     

      tar -zxvf cudnn-7.0-linux-x64-v4.0-prod.tgz
     
    
   

    

     

    
    

     

      cd cuda
     
    
   

    

     

    
    

     

      sudo cp lib64/* /usr/local/cuda/lib64/
     
    
   

    

     

    
    

     

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

6、测试cuda的samples

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
make
sudo ./deviceQuery

5、执行

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6 #删除原有动态文件
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成软链接
sudo ln -s libcudnn.so.6 libcudnn.so #生成软链接

sudo ldconfig

注意！！软链接中的数字6.0.21等参照文件夹lib64中原有的软链接名

设置环境变量，在/etc/profile中添加CUDA环境变量，

sudo gedit /etc/profile
 

在打开的文件中加入如下两句话

  

   

    

     

    
    

     

      export PATH=/usr/local/cuda/bin:$PATH
     
    
   

    

     

    
    

     

      export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
     
    
  
 

3、python

输入python后输入以下示例

>>>import tensorflow as tf
>>>hello = tf.constant('Hello, TensorFlow!')
>>>sess = tf.Session()
>>>print sess.run(hello)
Hello, TensorFlow!
>>>a = tf.constant(10)
>>>b = tf.constant(32)
>>>print sess.run(a+b)
42
>>>

5、测试成功接下来首先退出python  按快捷键Ctrl+D

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/79854075