Windows10中用Anaconda安装tensorflow-gpu(2.1.0)+报错:cudaGetDevice() failed. Status: CUDA driver version...
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, 如下: