pip安装deb_趟坑:使用pip安装TensorFlow

这几天在安装TensorFlow,看了很多教程,方法也试了几种。

最后还是用pip安装成功的,过程如下。

1.安装ubuntu后在  系统设置-软件与更新-附加驱动 里,更新N卡驱动。

(N卡官网下载run格式驱动手动安装比较麻烦,需要关闭X server,并禁止系统自带的开源驱动)

按照提示,重启。重启之后,终端输入nvidia-smi,可以看到GPU的信息。

2.N卡开发者官网下载CUDA deb格式安装包,这里注意,CUDA、cudnn、TensorFlow之间版本有要求。我最后使用的是cuda8.0+cudnn6.0+TensorFlow1.4

下载cuda的deb格式安装包,然后安装

#进入下载文件所在目录,执行下列命令:

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

3.N卡开发者官网下载cudnn,需要注册并填写一个问卷。

#进入下载文件所在目录,执行下列命令:

tar xvzf cudnn-8.0-linux-x64-v6.0.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

#退到根目录,运行下面语句:

sudo chmod a+r /usr/local/

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值