CentOS 7 安裝cuda環境

建立時間:2016-11-25html

做者:海濱python

背景介紹:爲了加速類prisma風格圖片渲染速度(開源項目yusuketomoto/chainer-fast-neuralstyle),先須要對一臺裝有GeForce GTX 780 Ti的CentOS機器安裝cuda環境。linux

安裝CentOS 7系統

  • CentOS官網下載安裝鏡像,本次下載的是Minimal ISO版本(不帶圖形化界面700多M)
  • 在mac終端使用dd命令製做啓動盤:
    1. 查看全部硬盤:diskutil list
    2. 取消硬盤掛載:diskutil unmountDisk /dev/disk2
    3. 拷貝iso鏡像文件(時間較久請耐心等待,速度2m/s):sudo dd if=CentOS-7.0-1406-x86_64-DVD.iso of=/dev/disk2 bs=1m
    4. 彈出硬盤:diskutil eject /dev/disk2
  • 啓動電腦從U盤啓動安裝系統(tips:若是已有windows系統,只需將系統安裝至空白分區便可

網絡配置

因爲ifconfig命令沒有安裝,能夠使用ip命令代替。git

顯示當前網卡信息:ip addr showgithub

編輯網卡配置:vi /etc/sysconfig/network-scripts/ifcfg-eno1bootstrap

新增IP地址和網關信息:windows

IPADDR=192.168.199.88
GATEWAY=192.168.199.1
DNS1=192.168.199.1

將ip地址獲取從dhcp改爲static:centos

BOOTPROTO=static

重啓網絡服務:service network restart 測試網絡服務:ping www.baidu.com服務器

前提軟件安裝

pip安裝 當前系統自帶python卻沒有pip,暈!網絡

curl "https://bootstrap.pypa.io/get-pip.py" -o "get-pip.py」
python get-pip.py

其餘軟件

  • git, sudo yum install git
  • gcc, sudo yum install gcc
  • g++, sudo yum install gcc-g++
  • kernel開發環境(編譯cuda須要), sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
  • python開發環境(編譯chainer須要), sudo yum install python-devel

安裝cuda

首先下載cuda安裝包,本次下載的是CentOS 7 runfile版本

參照官方手冊操做,進行安裝便可,這裏大體介紹下本身安裝過程當中的坎。

  • 安裝gcc、g++、kernel開發環境
  • 關閉系統自帶驅動Nouveau drivers(官方手冊有詳細方法)
  • 安裝cuda
  • 編譯cuda samples,測試安裝結果,運行./NVIDIA_CUDA-8.0_Samples/bin/x86_64/linux/release/deviceQuery

安裝cudnn

使用GPU加速神經網絡的計算 下載cudnn安裝包,其實就是lib庫和頭文件的壓縮包,最終解壓拷貝到cuda安裝路徑的lib6四、include文件便可。

安裝風格圖片渲染環境chainer

參照chainer官方文檔

加速結果測試

服務器配置:

CPU: Intel(R) Xeon(R) CPU E3-1270 v3 @ 3.50GHz(8核)

GPU: GeForce GTX 780 Ti

不使用GPU加速

運行代碼:python generate.py sample_images/tubingen.jpg -m models/composition.model -o sample_images/output.jpg 運行時間:13s

使用GPU加速

運行代碼:python generate.py sample_images/tubingen.jpg -m models/composition.model -o sample_images/output.jpg -g 0 運行時間:1.5s

加速效果很是明顯,性能提高接近10倍。

相關文章
相關標籤/搜索