我使用以下命令将带有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中引发LinAlgError("SVD没有收敛")numpy.linalg .linalg.LinAlgError:SVD没有收敛
其他相关消息来源建议检查NaN和Infs的数据,但我的数据肯定是干净的.顺便说一句,CPU版本的安装工作正常,只有在尝试在GPU上运行时才会出现问题
我试图重新安装Anaconda,重新安装CUDA和numpy,但它没有成功.