pip安装tf-gpu=2.4的bug解决方案

在CentOS7.9服务器上,使用conda安装tensorflow-gpu 2.4失败,提示包不可用。尝试使用pip安装后,虽然成功但测试发现无法使用GPU。错误提示缺少libcusolver.so.10和libcudnn.so.8。通过查找并复制缺失库到指定目录解决了问题,现在GPU可以正常使用。
摘要由CSDN通过智能技术生成

背景:在centos7.9服务器上,v100单卡,cuda-11.7,驱动也是匹配的,安装torch不费劲,但是tf-gpu就是不行,conda安装失败,只有pip安装,然而pip失败。woc

conda安装失败:

conda install tensorflow-gpu==2.4

Collecting package metadata (current_repodata.json): done

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

Collecting package metadata (repodata.json): done

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

PackagesNotFoundError: The following packages are not available from current channels:

  - tensorflow-gpu==2.4

Current channels:

  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64

  - h

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小李飞刀李寻欢

您的欣赏将是我奋斗路上的动力!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值