我安装了 CUDA 和 CUDNN,但仍然收到此错误

时间:2021-01-30 16:30:38

标签: tensorflow gpu tensorflow2.0 nvidia

<块引用>

导入错误:无法导入名称“multi_worker_mirrored_2x1_cpu” 'tensorflow.python.distribute.strategy_combinations' (C:\Users\astar\anaconda3\envs\Tensorflow 2\lib\site-packages\tensorflow\python\distribute\strategy_combinations.py)

对于 tensorflow=2.3.0,我需要 CUDA=10.2 和 CudNN=7.6。但是当我运行 TensorFlow GPU 来测试我是否安装了 Cuda 时,我收到了这个错误,有什么解决办法吗?

1 个答案:

答案 0 :(得分:1)

支持 GPU 的 Tensorflow 2.3.0 需要 CUDA 10.1 和 cuDNN 7.6。卸载版本 10.2 并安装兼容版本。

https://www.tensorflow.org/install/source#linux