本站教程收集整理的这篇文章主要介绍了从零开始安装TensorFlow1.0+keras(Ubuntu16.04+CUDA8.0+Cudnn5.1+TITANX),本站教程本站觉得挺不错的,现在分享给大家,也给大家做个参考。
NVIDIA TITANX 显卡安装
官网下载NVIDIA驱动
点击进入驱动下载地址
按照需求选择合适的GPU,因为我们使用的是TITAN X GPU 所以选择如下,操作系统旋转Linux 64-bit,点击搜索,点击下载后进入如下页面。
在这里要注意,不要点击同意并开始下载,要右键另存为!!!
至此,驱动程序便下载成功。
关闭 X server
(2) 如果/etc/init.d/ 目录下存在lightdm
显卡驱动安装完成以后,对应于gdm/lightdm使用如下命令来重启 X server
killall X不可使用,因为虽然停止了X server,但是gdm/lightdm会重新启动X server,其他C+A+B组合键也不可以,必须停止X server的父进程,否则还会不断生成X server
安装NVIDIA驱动
安装NVIDIA驱动之前,我们需要彻底卸载已经安装好的
sudo apt-get remove --purge nvidia-*
sudo apt-get install ubuntu-desktop
sudo rm /etc/X11/xorg.conf
echo 'nouveau' | sudo tee -a /etc/modules
如果之前没有安装过NVIDIA显卡,执行上面的命令可能提示找不到目录或者文件,可以忽略。
进入驱动所在的目录,执行以下命令进行安装
sudo sh NVIDIA-Linux-*.run
CUDA8.0 安装
点击进入CUDA 下载网址,选择合适版本下载,为*.run文件
安装完毕后验证安装
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
测试代码,如果可以打印如下页码,则代码安装成功
CUDNN5.1 配置
安装CUDNN相对简单,首先注册NVIDIA的开发账号,然后才能下载CUDNN
下载下来后
tar -zxf cudnn-8.0-linux-x64-v5.1.tgz
cd cuda
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
更新系统
sudo apt-get update && sudo apt-get upgrade
sudo apt-get install linux-source
sudo apt-get install linux-headers-`uname -r`
anaconda安装
香港vps
点击进入anaconda下载地址
对应自己python版本下载,比如我使用的2.7版本,进入下载文件目录运行安装命令
bash Anaconda2-4.3.1-Linux-x86_64.sh
然后各种下一步就可以了,最后它会问你用不用把它加入环境变量,这里最后选yes,然后执行
source ~/.bashrc
更新环境变量,以后你使用的python解释器就是anaconda里面的了,不再是系统自带的了。
tensorflow 安装
科普一下:现在最流行的Python集成环境是anaconda ,里面集成了上百个常用的python包,省的安装一大堆python的包了。anaconda里面自带了一个pip。如果你在终端使用的是sudo pip install,那么你安装的python包会安装到系统的python里面,也就是/usr/bin/python2.7里面,如果你使用的是pip install,那么你安装的软件包是安装到anaconda里面的python环境。因为我使用的是anaconda,所以使用pip而不是sudo pip
点击进入tensorflow1.0下载页面
@H_674_142@选择合适的版本,我们选择python2.7下的GPU版本
由于我们利用anaconda来安装tensorflow,所以只需要
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl
如果你想给系统自带的python安装tensorflow,只需要在前面加一个sudo就可以了
执行上图中的命令,出现类似的结果就代表tensorflow已经安装成功。
keras 安装
- 通过执行命令,来直接安装keras
pip install keras
- 从github下载keras
git clone https://github.com/fchollet/keras.git --recursive
可以看到TITAN X 的速度还是相对较快的,一次epoch 大概25秒左右。
本站总结
以上是本站教程为你收集整理的从零开始安装TensorFlow1.0+keras(Ubuntu16.04+CUDA8.0+Cudnn5.1+TITANX)全部内容,希望文章能够帮你解决从零开始安装TensorFlow1.0+keras(Ubuntu16.04+CUDA8.0+Cudnn5.1+TITANX)所遇到的程序开发问题。
如果觉得本站教程网站内容还不错,欢迎将本站教程推荐给好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。