由于科研需要本人入坑了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
在该环