Windows10中用Anaconda安装tensorflow-gpu（2.1.0）+报错：cudaGetDevice() failed. Status: CUDA driver version...

最新推荐文章于 2023-12-14 16:11:55 发布

饭饭饭小Jian

最新推荐文章于 2023-12-14 16:11:55 发布

阅读量645

点赞数 1

文章标签： anaconda

本文链接：https://blog.csdn.net/weixin_44287435/article/details/106097438

版权

Windows10中用Anaconda安装tensorflow-gpu（2.1.0）

1.安装Anaconda,添加环境变量；

anaconda官网 https://www.anaconda.com/distribution/
在这里插入图片描述

2. 打开Anaconda prompt, 配置环境安装tensorflow

1. 添加国内源，加速：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

2. 显示源地址：

conda config --set show_channel_urls yes

3. 安装tensorflow GPU版本到自己的虚拟环境：

conda 虚拟环境创建：
conda create -n tf2 tensorflow-gpu # 建立名为tf2的conda虚拟环境
conda activate tf2 # 激活名为tf2的Conda虚拟环境

另外有这些：
conda deactivate # 退出当前的Conda虚拟环境
conda env remove -n tf2 # 删除名为tf2的Conda虚拟环境
conda env list # 列出所有Conda虚拟环境

4. 将ipython：

where python：发现python不在envs\tf2里，所以在tf2环境中改过来：
pip install ipython

5. 测试当前gpu版本是否可用

ipython
import tensorflow as tf
tf.test.is_gpu_available() 在这里插入图片描述

6. 前一步若报错的处理

如果出现报错：cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
则是因为cuda和显卡驱动不一致，或者TensorFlow版本与cuda版本有冲突，搜了半天如何匹配，最后我的解决办法是：

a. 在设备管理器中核对电脑的GPU型号：

在这里插入图片描述

b. 在英伟达网站中选择对应的GPU驱动下载安装，例如我的：

在这里插入图片描述

c. 在anaconda navigator - environment项目下，在创建的tensorflow环境中，搜索cud，选中cudatoolkit 和 cudnn，点击右下角的apply。

这样子就配置好了，最后匹配的版本是：tf2.1-gpu — cuda 10.2, 如下：
在这里插入图片描述