安装TensorFlow时libcudart版本问题

问题描述

我刚刚在一个全新的conda环境中安装了tf，并且尝试安装了两个版本的cudnn，不确定是否相关。GPU不工作，因为我收到此错误：

tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory

推荐答案

请创建虚拟环境以在蟒蛇中安装TensorFlow。

按照以下代码在虚拟环境中安装TensorFlow：

conda create -n tf tensorflow  #Create a Virtual environment(tf).
conda activate tf              #Activate the Virtual environment
conda install tensorflow-gpu   #install TensorFlow-gpu in it.

运行conda install tensorflow-gpu时，这将自动在您的系统中安装所有必需的库集和软件包以及兼容版本的CUDA, cuDNN。

如果要安装特定版本的TensorFlow-GPU，可以执行以下代码：

conda install tensorflow-gpu==2.3.0  # specified version

注意：每次要使用TensorFlow时都需要激活虚拟环境。

请检查this链接以了解TensorFlow内部版本配置详细信息。

这篇关于安装TensorFlow时libcudart版本问题的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持跟版网！