安装Ubuntu+gpu+tensorflow+py2+py3

最新推荐文章于 2023-04-06 02:13:12 发布

xiongvvvxiong

最新推荐文章于 2023-04-06 02:13:12 发布

阅读量537

点赞数

CC 4.0 BY-SA版权

文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/xiongvvvxiong/article/details/77692849

本文详细介绍了在Ubuntu 16.04系统上安装CUDA 8.0和CUDNN 6.0的过程，包括U盘分区设置、NVIDIA显卡驱动安装、CUDA及CUDNN的安装步骤和环境变量配置等关键信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一，首先要准备一个够用的U盘 >2g，安装过程中需要分区，参考以下两个教程两块硬盘分了三个区，首先根目录/（EXT4, 主分区）预留了190g也就是ssd剩下的容量,其中swap预留了64g和内存同等（交换空间，逻辑分区），最后4T的全部留给了/home（EXT4, 逻辑分区）。

https://zhidao.baidu.com/question/181607133.html

http://www.ah168.com/jishuwenzhang/xitong/Ubuntu-916.html

二，安装显卡驱动：参考以下教程，需要注意的是需要从官网上直接下载，此处下载的版本是NVIDIA-Linux-x86_64-384.69.run，安装时需要用命令

sudo chmod a+x NVIDIA-Linux-x86_64-375.20.run

sudo ./NVIDIA-linux-x86_64-375.20.run –no-x-check –no-nouveau-check –no-opengl-files

需要说明的是不要从界面中安装，也不要从cuda中自带的安装（下载run文件，过程中跳过显卡安装）。安装过程中在Kernel 处选择No, 32bit compatinity处选择yes，而在X config 改动处选择No(基本都是默认选项)，另外需要说明的是若通过其他方式安装导致无法登陆界面时可以采用命令 sudo apt-get remove –purge nvidia* 卸载安装驱动。

http://blog.youkuaiyun.com/u010159842/article/details/54344683

三，下载Cuda 8.0 for ubuntu16.04 以及cudnn 6.0 for cuda8 (现在已经是cuda 9 与cudnn 7了，但为了稳定，先不更新, 另外Cuda有一个补丁也先不更新(应该是覆盖)，之后需要时再更新，注意cudnn的版本6 还是5.1在下一篇博客中说明)。Cuda 的安装命令

sudo sh cuda_8.0.61_375.26_linux.run，不再选择安装驱动，其余都可选yes和默认。之后把cudnn的文件进行覆盖。此时需要更改环境变量（bashrc,明天再说，另外网上说需要降低gcc版本，目前没有遇到问题，遇到问题后再看如何降低版本），以下两个总参考教程：

http://blog.youkuaiyun.com/zhaoyu106/article/details/52793183/

http://www.52nlp.cn/%E6%B7%B1%E5%BA%A6%E5%AD%A6%E4%B9%A0%E4%B8%BB%E6%9C%BA%E7%8E%AF%E5%A2%83%E9%85%8D%E7%BD%AE-ubuntu-16-04-nvidia-gtx-1080-cuda-8

四，在安装完Cuda以及将cudnn复制后，此时需要配置下环境变量，为了多用户考虑，可以采用如下的配置方法：

sudo gedit .bashrc

export PATH=/usr/local/cuda-8.0/bin:$PATH  
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

export LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

source .bashrc

最后 sudo ldconfig -v使其生效到目前为止CUDA与CUDNN安装完毕，接下来需要测试是否安装成功：进入/usr/local/cuda/samples, 执行下列命令来build samples
  
   sudo make all -j8 

  
全部编译完成后， 进入/usr/local/cuda/samples/1_Utilities/deviceQuery$ ./deviceQuery , 运行deviceQuery
  
   /usr/local/cuda/samples/1_Utilities/deviceQuery$ ./deviceQuery 
如果出现显卡信息以及PASS成功（此处安装的driver version = 9.0, runtime version=8.0, 应该可以，可能会有少许速度上影响，反过来就不行了，）以上cuda教程参考如下（不需要考虑其中的CUDNN）:
http://www.cnblogs.com/jinggege/p/5766146.html

 需要说明的是，在某些情况下可以安装两个版本的cuda,此时不要链接文件cuda,保留cuda原来的版本号，进而在.bashrc中将地址路径都改为带版本号的cuda地址。