Linux系统下安装TensorFlow的GPU版本(亲测可行)

安装TensorFlow-GPU在Linux环境下可能遇到诸多问题,关键在于匹配正确的CUDA Toolkit版本、cuDNN版本以及Python版本。确保Driver Version与CUDA Toolkit对应,从NVIDIA官网获取历史版本。cuDNN下载时注意版本号,以v开头的才是实际版本。安装tensorflow-gpu-1.4时,可使用pip3,如果速度慢,可指定TensorFlow的GPU wheel包URL进行升级安装。详细步骤见指定博客链接。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

安装tensorflow-gpu真的好烦,个人经验来说,最主要的就是找好一下对应关系(我的是python3.5)

1 Linux x86_64 Driver Version与CUDA Toolkit的对应,一定要对应好,CUDA超出了Driver Version的话,是无法使用的

https://developer.nvidia.com/cuda-toolkit-archive

CUDA与Driver Version对应关系

2 CUDA与cuDNN的对应关系

CUDA与cuDNN

 3 在下载cuDNN的时候要注意了,链接https://developer.nvidia.com/rdp/cudnn-archive 不要看cudnn-(这个数字),应该以v(这个才是版本号).tgz

4 在安装好CUDA和cuDNN之后安装tensorflow,一定要对应好版本号

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值