ubuntu16.04下2080Ti显卡配置 cuda10.0 + cudnn7.4.2 + tensorflow 1.13.1

ubuntu16.04下2080Ti显卡配置cuda10.0+cudnn7.4.2 + tensorflow 1.13.1 + pytorch 1.0.0 + pytorch 1.0.0

友情提醒，没有一个环境配置只看一个文档就能解决的，所以还是需要多百度，多谷歌。。。

• 下载相关文件
  • 下载cuda，cuda_10.0.130_410.48_linux.run
  • 下载cudnn，cudnn-10.0-linux-x64-v7.4.2.24.tgz
  • 下载TensorRT
  • 下载pytorch,(链接: https://pan.baidu.com/s/1p-8gR8xGWYMj4FBCW1uPzQ 提取码: 6666)，其他版本似乎有问题
• 安装cuda
  • 如果之前安装过cuda需要先删除，运行

     sudo apt-get purge nvidia-cuda* 
    

  • 将默认的nouveau显示驱动加入黑名单

    sudo vim /etc/modprobe.d/blacklist.conf
    

    • 追加

      blacklist nouveau
      options nouveau modeset=0
    

    • 保存并退出
  • 离开图形界面，进入命令行模式，

    Ctrl+Alt+F1进入命令行模式，Ctr+Alt+F7表示返回图形操作界面

    • 禁用X-server

      sudo service lightdm stop
      

    • 更新修改的黑名单

      sudo update-initramfs -u
      

    • 安装cuda

      cd ~/Downloads
      sudo ./cuda_10.0.130_410.48_linux.run
      

      因为之前没有安装显卡驱动，所以这里要安装显卡的图形驱动【省去别的地方找显卡驱动】

    • 安装完成后如果报安装不完整，需要安装cuda driver，则需要根据提示进行安装
    • 修改配置文件

      sudo vim /etc/profile
      # 增加下面几行
      export PATH=/usr/local/cuda-10.0/bin:$PATH
      export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
      export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
      export CUDA_HOME=/usr/local/cuda-10.0:$CUDA_HOME
      

    • source一下，然后输入以下命令检测以下版本

      nvcc -V
      

    • 重启
• 配置cudnn

  解压cudnn-10.0-linux-x64-v7.4.2.24.tgz，解压后得到文件夹cuda，当前路径仍然为~/Downloads，进行相关文件的配置

  sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
  sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
  sudo chmod a+r /usr/local/cuda/include/cudnn.h 
  sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
  

  • 查看cudnn版本

    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    

    即可看见7 4 2

• pip换源

  vim ~/.pip.conf	  
  # 添加下面几行
  [global]
  index-url = https://pypi.tuna.tsinghua.edu.cn/simple
  [install]
  trusted-host=mirrors.aliyun.com
  

• 安装tensorflow1.13.1

  pip install tensorflow-gpu==1.13.1
  

• 安装pytorch

  pip install ~/Downloads/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
  

• 参考
  • Ubuntu16.04 RTX2080ti 安装NVIDIA驱动+cuda10.0+cudnn7.3.1
  • How to install Nvidia drivers and cuda-10.0 for RTX 2080 Ti GPU on Ubuntu-16.04/18.04
  • tensorflow与cuda、cudnn匹配版本