tensorflow安装

    准备在实验室电脑上使用tensorflow仿真。由于之前已经安过cuda7.5,现在不够用了要升级,一步步来。

1.升级cuda7.5到cuda8.0,安装cudnn5.1

首先卸载cuda7.5

cd /usr/local/cuda/bin
sudo ./uninstall_cuda_7.5.pl
       在System Settings->Software & Updates->Additional Drivers把显卡的最新的闭源驱动装了,重启之后正常的话,显卡驱动就是装好了。cuda 8.0下载之后是一个.run文件,给它用chmod +x添加可执行属性之后用sudo运行,别安装里面的驱动,只装cuda tookits。

  下载cudnn,要注册一个账号,看清楚了Tensorflow目前最新版本是1.1.0-rc0,需求cuda 8.0和cudnn 5.1 (5.0不行)。cudnn 5.1下载以后解压可以看到cuda文件夹,按照下面命令把相应头文件和库文件拷贝到系统路径下面就行了。

cd cuda 
sudo cp include/*.h /usr/local/cuda/include/  (注意这里是你自己的cuda的安装地址,复制*.h文件) 
sudo cp lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
2.使用pip安装tensorflow

pip install tensorflow_gpu-1.1.0rc2-cp27-none-linux_x86_64.whl 




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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值