ubuntu kylin 16.04 安装tensorflow-gpu-1.4

截止2018.2.21日,tensorflow1.5并不能支持最新版本的cuda9.1

  • 因此,想要安装gpu版本需安装tensorflow1.4版本(支持cuda8)
sudo python3 -m pip install tensorflow-gpu==1.4
  • CUDA8目前有两个版本,GA2(发布于2017年2月)和GA1(发布于2016年9月)

我选的是GA2,runfile(local) --- deb(local)文件在安装时会无法选择版本,在安装好配置文件后,会默认安装最新版本



        安装过程中会提示是否需要安装驱动(我选的文件默认安装驱动375.26),你可以打开自己的系统设置-软件更新-附加驱动,里面查看自己是否有NVIDIA的驱动(cuda8支持365以上版本)。如果你已有显卡驱动,就不要再重复安装别的版本驱动。

  • cudnn对应cuda8也有两个版本,要选择cudnn v6.0,不然在import tensorflow时会出现(ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory)错误


        安装时要安装cuDNN v6.0 Runtime 和 Developer 两个文件,(code samples可选安装)

  • 在使用过程中,如果出现ImportError: libcudnn.so.8: cannot open shared object file: No such file or directory,需要
    在环境变量中加入
    export PATH="$PATH:/usr/local/cuda-8.0/bin"
    export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"
    二者缺一不可(这里的路径是安装时候的默认路径,具体路径因人而异)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值