我使用以下命令将具有 GPU 支持的 Tensorflow 和 Keras 安装到 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 https://stackoverflow.com/questions/24173301/numpy-polyfit-suppress-printing-of-intel-mkl-error-message建议检查 NaN 和 Infs 的数据,但我的数据肯定是干净的。顺便说一句,CPU 版本的安装工作正常,仅当尝试在 GPU 上运行时才会出现此问题
我尝试安装 Anaconda、安装 CUDA 和 numpy,但没有成功。
问题出在包上MKL (2018.0.0)- 它似乎是最近发布的,并且与通过 conda* 随 Tensorflow(1.3.0) 和 Keras(2.0.5) 提供的某些软件包的版本冲突。
所以我手动降级了mkl使用 Anaconda Navigator 到 v11.3.3,这会自动降级其他软件包,现在一切正常。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)