【发布时间】:2018-03-22 01:05:45
【问题描述】:
我使用以下命令将支持 GPU 和 Keras 的 Tensorflow 安装到 Anaconda (v1.6.5) 的环境中:
conda install -n EnvName tensorflow-gpu
conda install -n EnvName -c conda-forge keras-gpu
我的机器上有 NVIDIA Quadro 2200K 驱动程序 v384.66、cuda-8.0、cudnn 7.0
当我在训练阶段尝试使用 Keras 运行 python 代码时,我得到以下信息
英特尔 MKL 错误:进入 DLASCL 时参数 4 不正确。
以后
文件 “/home/User/anaconda3/envs/keras_gpu/lib/python3.6/site-packages/numpy/linalg/linalg.py”, 第 99 行,在 _raise_linalgerror_svd_nonconvergence raise LinAlgError("SVD 没有收敛") numpy.linalg.linalg.LinAlgError: SVD 没有收敛
其他相关sources 建议检查 NaN 和 Infs 的数据,但我的数据肯定是干净的。顺便说一句,安装的 CPU 版本运行良好,仅在尝试在 GPU 上运行时才会出现此问题
我尝试重新安装 Anaconda,重新安装 CUDA 和 numpy,但没有成功。
【问题讨论】:
标签: numpy tensorflow keras anaconda intel-mkl