
对压缩包进行解压,得到一个名为cuda的文件夹。正常来说,我们需要将cuda中的一些文件复制到/usr/local/cuda-8.0/***的路径下,并配置相关路径。但这里,我们的做法有点不同。
因为,我们是用apt-get安装的,所以,我们是不存在/usr/local/cuda-8.0/***路径的,也不需要配置环境路径。但复制还是要复制的,不过位置变了一下,这里是第二个卡住我的问题点,找了很久,知乎的一位答友吕俊奇的回答给了我启发(回答的链接我会放到文末的参考中),这里要做如下操作:
第一,将解压出来的cuda/include/下的cudnn.h文件复制到/usr/local/include/目录下。
第二,将cuda/lib64/目录下的所有文件复制到python环境的tensorflow包的tensorflow/python/目录下,比如,对我来说该路径就是/home/aaron/.local/lib/python2.7/site-packages/tensorflow/python,这个路径因人而异,请参考我的路径自行查找。
至此,安装全部完成。我们测试一下是否安装成功。
在测试之前,埋一下前面的坑。
关于bumblebee版显卡驱动在深度学习训练中的开启和关闭:
在进行训练前,在终端输入sudo tee /proc/acpi/bbswitch <<< ON,可开启显卡。
在训练结束后,在终端输入sudo tee /proc/acpi/bbswitch <<< OFF,可关闭显卡。
开启显卡在终端中打开python,并尝试引入tensorflow:
python
>>> import tensorflow as tf
如果没有安装成功,会报错。
如果没有报错,那说明安装的没有问题。继续输入:
>>> sess=tf.Session()
会输出显卡信息:
好了,恭喜,到了这里,你已经完成了全部内容,祝你好运!
[1] Deepin15.4 下 CUDA 配置方法
[2] deepin15.5下安装tensorflow
[3] Ubuntu16.04无法安装CUDA吗? - 吕俊奇的回答 - 知乎
本文来自电脑杂谈,转载请注明本文网址:
https://www.pc-fly.com/a/shouji/article-81404-2.html
此后