【环境搭建】ubuntu18.04+CUDA10.1+cuDNN7.6.4+TensorFlow1.15.0

一、WIN10 + ubuntu18.04双系统安装

  1. rufus制作系统安装盘,注意选GPT类型
  2. 关闭WIN10快速启动以及BIOS的SECURE BOOT选项
  3. BIOS选择U盘启动项,进入ubuntu安装界面
  4. 只需自定义/分区和HOME分区,注意引导要写入WINDOWS BOOT MANAGER,方便从开机从UEFI选择进入ubuntu还是win10系统,而不是依赖于grub引导
  5. 系统重装,推荐使用easyUefi,先恢复WIN10引导,然后删除ubuntu所在分区卷,这样比较保险

二、CUDA10.1安装

  1. NVIDIA官网测评显卡型号,决定驱动程序版本和CUDA版本
  2. CUDA的安装推荐使用apt-get方式,会连同显卡驱动一并安装,不推荐先手动安装显卡驱动,后续如果发现显卡驱动版本和CUDA版本不适配,显卡驱动会很难卸载干净,最后只能重装系统
  3. CUDA安装完成后,重启电脑,nvidia-smi命令验证显卡驱动版本,nvcc -V验证CUDA版本

三、cuDNN配合CUDA版本安装

  1. 选对版本,下载压缩包
  2. 解压到/usr/local,主要是往CUDA目录添加库文件
  3. 添加环境变量到~/.bashrc,添加路径到/etc/ld.so.conf,sudo ldconfig,重启电脑,这步很关键,TensorFlow容易找不到CUDA库文件

四、Anaconda安装

  • 主要是为了安装Python虚拟环境方便,conda create命令创建虚拟环境,指定Python版本

五、TensorFlow安装

  • TensorFlow官网教程很详细,但是下载速度非常慢,建议找国内的镜像,关键就是匹配前一步的Python版本,指定gpu版以及选对版本,pip命令安装即可

你可能感兴趣的:(【环境搭建】ubuntu18.04+CUDA10.1+cuDNN7.6.4+TensorFlow1.15.0)