不能用gpu跑,一直用cpu跑模型引发的一些问题,服务器安装不上tensorflow-gpu

在Linux服务器上尝试安装TensorFlow-GPU时遇到问题,即使设置了CUDA_VISIBLE_DEVICES,模型仍然在CPU上运行。尝试使用pip安装1.12.0版本报错,疑似CUDA版本不匹配,但检查后CUDA版本无误。之后改用conda安装1.6版本,虽然能导入TensorFlow,但在运行程序时仍提示环境未安装好。删除并重建环境,甚至删除镜像,问题仍未解决,导致存在3个TensorFlow版本。问题可能与镜像或conda环境有关。

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

先是死活总是在cpu上,实际上用了os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' 还是一样的 

就是一直在cpu上跑,真的是让人生气。

 

然后有人建议用pip 安装 ,用pip install tensoflow-gpu  自然安装的是1.12.0

但是我发现报错了

话说这是cuda版本不对 ,这个我就查了一下 cuda版本  发现没毛病呀 

 

最后我的自觉告诉我还是不要使用pip安装 不知道为什么 因为之前也在这太服务器上使用过pip但是就是会报错的

所以我还是按照conda安装  

这次我使用了1.6  conda的时候出现要安装很多东西

我觉得有戏哈  ,但

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值