【环境搭建】ubuntu18.04+CUDA10.1+cuDNN
2020-02-23 本文已影响0人
无为无悔
一、WIN10 + ubuntu18.04双系统安装
- rufus制作系统安装盘，注意选GPT类型
- 关闭WIN10快速启动以及BIOS的SECURE BOOT选项
- BIOS选择U盘启动项，进入ubuntu安装界面
- 只需自定义/分区和HOME分区，注意引导要写入WINDOWS BOOT MANAGER，方便从开机从UEFI选择进入ubuntu还是win10系统，而不是依赖于grub引导
- 系统重装，推荐使用easyUefi，先恢复WIN10引导，然后删除ubuntu所在分区卷，这样比较保险
二、CUDA10.1安装
- NVIDIA官网测评显卡型号，决定驱动程序版本和CUDA版本
- CUDA的安装推荐使用apt-get方式，会连同显卡驱动一并安装，不推荐先手动安装显卡驱动，后续如果发现显卡驱动版本和CUDA版本不适配，显卡驱动会很难卸载干净，最后只能重装系统
- CUDA安装完成后，重启电脑，nvidia-smi命令验证显卡驱动版本，nvcc -V验证CUDA版本
三、cuDNN配合CUDA版本安装
- 选对版本，下载压缩包
- 解压到/usr/local，主要是往CUDA目录添加库文件
- 添加环境变量到~/.bashrc，添加路径到/etc/ld.so.conf，sudo ldconfig，重启电脑，这步很关键，TensorFlow容易找不到CUDA库文件
四、Anaconda安装
- 主要是为了安装Python虚拟环境方便，conda create命令创建虚拟环境，指定Python版本
五、TensorFlow安装
- TensorFlow官网教程很详细，但是下载速度非常慢，建议找国内的镜像，关键就是匹配前一步的Python版本，指定gpu版以及选对版本，pip命令安装即可
