我使用以下命令将带有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没有收敛
其他相关sources建议检查NaN和Infs的数据,但我的数据肯定是干净的。顺便说一下,安装的CPU版本工作正常,只有在尝试在GPU上运行时才会出现问题
我尝试重新安装Anaconda,重新安装CUDA和numpy,但它没有用完。