TensorFlow-GPU版本的选择与安装(国内源)

cuda和cudnn的选择

cuda的选择

TensorFlow1.0至1.4版本只支持CUDA8.0
TensorFlow1.5至1.12版本支持CUDA9.0
TensorFlow1.13之后的版本,支持10.0
更多版本CUDA:
https://developer.nvidia.com/cuda-toolkit-archive

cuDNN

cuDNN库的版本选择也是有规定的。以Windows10操作为例:
TensorFlow1.0到1.2版本使用的是cuDNN5.1版本(安装包为cudnn8.0-Windows10-x64-v5.1.zip)
TensorFlow1.3到1.4版本使用的是cuDNN6.0版本(安装包为cudnn8.0-Windows10-x64-v6.0.zip)
TensorFlow1.5到1.10版本使用的是cuDNN7.0版本(安装包为cudnn9.0-Windows10-x64-v7.rar)
TensorFlow1.11到1.12版本使用的是cuDNN7.2版本(安装包为cudnn9.0-Windows10-x64-v7.2.1.38.zip)
TensorFlow1.13之后使用的是cuDNN7.5版本(安装包为cudnn10.0-Windows10-x64-v7.5.056.zip)

TensorFlow-GPU国内源的安装

1.先安装anaconda创建环境

conda create -n tfgpu python=3.6.4

2.安装TensorFlowGPU1.13(清华源)

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.13.1

3.安装TensorFlowGPU2.0

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.0.0a0

注:清华镜像https://pypi.tuna.tsinghua.edu.cn/simple

  • 2
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值