您的位置:首页 > Web前端

踩坑分析(ubuntu 16.04 + NVIDIA GTX1070 + CUDA8.0 + cudnn8.0 + caffe2)

2017-11-12 08:53 676 查看
主题:caffe2环境配置

背景:搭建一个平台实践机器学习算法;贾扬清博士的盲粉,选择caffe2。

环境:

内存:16G

处理器:Intel® Core™ i7-7700 CPU @ 3.60GHz × 8

显卡:GeForce GTX 1070/PCIe/SSE2

系统类型:64-bit

硬盘:SSD 240G

本人为了实践机器学习,安装驱动我们不能再图形化终端中进行(不能一边骑驴一边杀驴),所以在安装时我们要进入字符终端(不需要显卡)。在这个过程中,字符终端可能显示不正常。在系统的核引导设置参数中加入“nomodeset”这个参数可以避免这个问题。如下所示:

$ sudo vi /etc/default/grub 或 sudo gedit /etc/default/grub //用文字编辑器打开grub设置文件

$ sudo update-grub // grub通过脚本grub-mkconfig更新/boot/grub/grub.cfg

edited: GRUB_CMDLINE_LINUX="nomodeset"

****************************

1. download driver file from Nvidia's website

2. install from repository, run these commands step by step:

    sudo apt-get remove nvidia*

    sudo add-apt-repository ppa:xorg-edgers/ppa

    sudo apt-get update

    sudo-get install nvidia-current

3. restart the computer

4. press ctrl+alt+F1, and login

5. run in terminal:

    sudo service lightdm stop    

    cd path-to-nvidia.run/path

    sudo chmod +x NVIDIA.run

    sudo service lightdm stop

    sudo run ./nvidia.run

6. sudo service lightdm start

7. restart your computer

8. run in the terminal: nvidia-smi 

9. OK, it's done
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息