美文网首页
【环境搭建】ubuntu18.04+CUDA10.1+cuDNN

【环境搭建】ubuntu18.04+CUDA10.1+cuDNN

作者: 无为无悔 | 来源:发表于2020-02-23 19:46 被阅读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+cuDNN

          本文链接:https://www.haomeiwen.com/subject/dmcsqhtx.html