【环境搭建】ubuntu18.04+CUDA10.1+cuDNN
作者:
无为无悔 | 来源:发表于
2020-02-23 19:46 被阅读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命令安装即可
本文标题:【环境搭建】ubuntu18.04+CUDA10.1+cuDNN
本文链接:https://www.haomeiwen.com/subject/dmcsqhtx.html
网友评论