cuda配置

Step 1 关闭第三方显卡驱动
打开Terminal终端(Ctrl+Alt+T)
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
(用编辑器打开blacklist文件,)
在文件中写入:(它是第三方驱动,回合cuda即官方驱动冲突)
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

在terminal中执行
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u
sudo reboot

Step 2 安装cuda
进入到cuda .deb安装位置(以下以7.5为例,)( Tab )
sudo dpkg -i cuda-repo-ubuntu1404_local_7.5-18_amd64.deb
sudo apt-get update
sudo apt-get install -y cuda

安装完毕之后,打开bashrc文件,将cuda路径写入:(sudo是要用cd在终端打开,gedit是文本编辑器)
sudo gedit ~/.bashrc

在bashrc文件中写入:(为了配置环境)
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=:/usr/local/cuda/lib64

Step 3 验证安装
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make –j16
sudo ./deviceQuery
运行测试代码,如果现在找到GPU则成功Result = PASS

Step 4 安装配置nVidia cuDNN 加速模型运算()
a. 安装前请去先官网下载最新的cuDNN (cudnn-7.0-linux-x64-v4.0-prod)。下载完成后直接双击安装包,把里面的文件CUDA拖到Downloads文件夹中,这一步很重要,要不然后面编译都不知道错在哪里了
$ cd ~/Downloads/CUDA
$ sudo cp include/cudnn.h /usr/local/include
$ sudo cp lib64/libcudnn.* /usr/local/lib

b. 链接cuDNN的库文件
$ sudo ln -sf /usr/local/lib/libcudnn.so.4.0.7 /usr/local/lib/libcudnn.so.4
$ sudo ln -sf /usr/local/lib/libcudnn.so.4 /usr/local/lib/libcudnn.so
$ sudo ldconfig –v

Step 5 Anaconda 安装(装python及其对应的依赖包,所以需要新开终端)
a. 安装前请去先官网下载最新的,直接打开Downloads文档
b. $ cd ~/Downloads
c. $ bash An….(文件名)
d. $ yes
e. $ sudo gedit ~/.bashrc (这一步配置环境变量)
f. 输入密码进入 文本 (为了修改环境变量) 复制 终端的( export ………………/bin/…….) 保存
g. 新开终端输入python 测试是否Anaconda安装成功

Step6 安装tensorflow
conda install tensorflow-gpu==1.2.0
安装完成后,现在终端中输入python
import tensorflow as tf
如果不报错,证明安装成功

Step7 安装keras
conda install keras
安装完成后,现在终端中输入python

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值