【已解决】Linux服务器安装Sionna 踩坑(Could not find cuda drivers on your machine, GPU will not be used)

由于科研需要本人入坑了Sionna,但是由于Sionna目前还比较小众,因此本人在安装和使用Sionna的过程中踩了不少坑,现在分享其中一个坑的解决办法。

Sionna本身是基于TensorFlow开发的,但目前版本号为'0.19.1'的Sionna依赖于TensorFlow 2.13-2.15和Python 3.8-3.11。如果你现有环境下的TensorFlow版本不在2.13-2.15之间,或者还未下载TensorFlow,则通过pip install sionna安装Sionna时会自动下载一个不含cuda的TensorFlow(2.13.1)并作为该环境下的默认TensorFlow来使用。但由于其不含cuda,因此在import sionna或import tensorflow时会出现如下警告:

I tensorflow/tsl/cuda/cudart_stub.cc:28] Could not find cuda drivers on your machine, GPU will not be used. 

这意味着你无法使用GPU对Sionna和TensorFlow进行加速计算。为了解决该问题,本人踩了无数坑,包括各种卸载/重装显卡驱动、cuda/cuDNN等,还成功把我在实验室服务器上原来的一个docker的显卡驱动搞坏,导致得重开一个docker来重新配环境+迁移数据了。。。不提这段血泪史了,下面分享一个简单的解决方法:

用conda新建一个环境

conda create -n sionna python=3.10
conda activate sionna

在该环

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值