ubuntu18.04 配置深度学习环境!

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/wuchangwen0216/article/details/88760581

在重装N+1次系统环境后,我终于学会了要整理文档。。。

首先,tensorflow支持cuda10,其次,个人之前也使用过caffe,所以也会有caffe和caffe2的教程。进入正题。

1. NVIDIA驱动配置

1.1 添加NVIDIA源

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
#更新源

1.2 查看可以更新的附加驱动

点击系统左下角菜单,进入软件和更新

选择附加驱动选项卡,更改为最新的NVIDIA驱动,然后点击应用更改

等待下方进度条完成后,重启电脑,显卡驱动安装完成!

注:BIOS里的防火墙要关闭,不然好像安不了驱动。因为博主很早前就关闭了,在这里就不复述了。

nvidia-smi
#查看显卡信息

2. 下载CUDA10, CUDNN7.5

版本不一定要一模一样,看个人的需求,下载相应的版本就行了。安装方法基本一样。

############################################################################################

待续

展开阅读全文

没有更多推荐了,返回首页