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

Ubuntu

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。