深度工具安装(NVIDIA+cuda+cudnn+tensorflow)

显卡驱动安装

查询nvidia驱动可用版本,运行sudo apt-cache search nvidia-* 查询相应版本

根据得到的结果,安装相应的版本。


2、官网下载显卡驱动包。

lspci | grep -i nvidia    (找自己机子的显卡型号)


  1. sudo add-apt-repository ppa:graphics-drivers/ppa  
  2. sudo apt-get update  
  3. sudo apt-get install nvidia-(版本号)

3、重启

  1. sudo shutdown -r now 

4、nvidia检查

  1. cat /proc/driver/nvidia/version  

CUDA安装

下载cuda最新版本:

  1. sudo dpkg -i (文件名).deb  
  2. sudo apt-get update  
  3. sudo apt-get install cuda  
添加cuda环境

  1. echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc  
  2. echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc  
  3. source ~/.bashrc 
 检查cuda版本

  1. nvcc -V  
重启

sudo shutdown -r now 


cudnn安装,选择下载的文件

  1. tar xvf cudnn*.tgz  
  2. cd cuda  
  3. sudo cp */*.h /usr/local/cuda/include/ 
  4. sudo cp */libcudnn* /usr/local/cuda/lib64/  
  5. sudo chmod a+r /usr/local/cuda/lib64/libcudnn*  
终端查看

  1. nvidia-smi 




Tensorflow安装(清华大学镜像)


sudo apt-get install python-pip python-dev


sudo pip install --upgrade pip

sudo pip install \
  -i https://pypi.tuna.tsinghua.edu.cn/simple/ \
  https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.2.0-cp27-none-linux_x86_64.whl


在sudo ldconfig时遇到

usr/local/cuda-8.0/lib64/libcudnn.so.5 不是符号连接的问题,解决办法也很简单,重新建立链接并删除原链接

首先找到usr/local/cuda-8.0/lib64/目录,搜索 libcudnn 然后发现

两个文件

libcudnn.so.5   和libcudnn.so.5.0.5  理论上只有一个libcudnn.so.5.0.5


终端执行

[python] view plain copy
  1. ln -sf /usr/local/cuda-8.0/lib64/libcudnn.so.5.0.5 /usr/local/cuda-8.0/lib64/libcudnn.so.5  

sudo ldconfig时就可以了,这时候会发现usr/local/cuda-8.0/lib64/目录下只有

libcudnn.so.5.0.5 文件了,libcudnn.so.5消失了。


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值