赞
踩
最近需要在服务器上配置tensorflow-gpu的环境来运行深度学习模型,以前在Windows上配置过,也知道一些注意点,这次在Linux下配置,也遇到了很多坑,下面总结一下配置过程,配置是使用Linux下安装的anaconda来进行的。
source activate tensorflow
nvidia-smi
conda install tensorflow-gpu==1.11.0
安装tensorflow-gpu1.11.0之后,conda会将cuda9.2.0、cudnn7.6.5进行安装,还有一些tensorflow的辅助库
此时就会出现一个问题,本机需要的cuda是9.1及以下的版本,但是使用conda进行安装要么安装了9.2版本以及以上版本的cuda(当安装tensorflow-gpu1.11.0及以上版本时),要么直接不给安装cuda以及cudnn(安装tensorflow-gpu1.10及以下版本时),而且cuda、cudnn单独安装非常麻烦,我就进行了一系列的uninstall cuda以及cudnn操作,做了很多无用功。
如果直接使用版本不对的cuda的话,运行深度学习模型则会出现这个错误
下面就来讲如何实现安装了cuda9.2版本之后,如何搞定这个难搞的问题
conda install -c numba cudatoolkit
实现了cudnn7.6.5,cuda9.0.0版本的安装
最终环境配置为
配置完成,可以利用GPU进行深度学习的训练
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。