GPU配置经验总结,希望对后面采同样坑者有所帮助
1.首先查看使用的机器是含有显卡、显卡更新和cuda对应版本信息的查询:
计算机——鼠标右键——设备管理——NVIDIA
强调一下:要确保NVIDIA是最新的,否侧后面会出现以下错误,当然也可能是其他因素引起的,其他因素引起的网上总结有两种:
方法一:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0"
其中0的位置你可以尝试设置成1
方法二:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
这是测试代码中的一项,
说回来,查看NVIDIA是否最新,接着上面,鼠标右键点击NVIDIA GeForce GT 740——属性—驱动程序——更新驱动程序
查看应该下载的cuda对应的版本
鼠标放在桌面空白处,鼠标右键单击——出现NVIDIA控制面板——进入
点击帮助——关于NVIDIA控制面板
2.下载cuda8.0和cudnn6.0
cuda8.0下载地址:https://developer.nvidia.com/cuda-80-ga2-download-archive
下载后进行安装,安装这个大家都懂,至于路径我的是默认安装就在c盘
安装好以后会在你c盘文件中:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
进行环境变量配置:
计算机——鼠标右键——属性——高级系统设置——高级——环境变量
其中CUDA_PATH和CUDA_PATH_V8_0是默认已经配好的,接下来你需要在系统变量中的PATH添加下面四个路径
就是刚刚提到路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下的:
1.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64;
2.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;
3.C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64;
4.C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64;
下载cudnn6.0:这个可能有点麻烦,需要耐心,慢慢等待,因为涉及到注册问题。
下载路径:https://developer.nvidia.com/cudnn
下载好以后会出现可以解压出一个cuda的文件,其中包括bin,include,lib,分别将其中最底层的文件复制到
此文件夹中对应的文件中即可
3.python3.6和pycharm的安装,网络上很多教程,大家搜索看看以及pycharm专业版的破解
4.pycharm中安装tensorflow-gpu
打开pycharm后,左边会有这样的一个文件:
因为之前试过pycharm中通过File——Setting——Project——Project Interpreter进行安装不可以,最后下载相应的tensorflow-gpu.whl文件放在Scripts进行安装:
安装好以后就可以进行测试啦:
import tensorflow as tf a = tf.constant([1.0, 2.0, 3.0], shape=[3], name='a') b = tf.constant([1.0, 2.0, 3.0], shape=[3], name='b') c = a + b # 通过log_device_placement参数来输出运行每一个运算的设备。 sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) print(sess.run(c)) |
出现结果:
到这里,恭喜你安装成功啦。
注意:安装过程中可能会出现
DLL load failed: 找不到指定的模块
请下载VC进行安装:https://pan.baidu.com/s/1Ocmch3p2pcvPfSSm9TxepA
希望对遇到同样问题的人有所帮助,第一次写博客,不足之处请多多指教,本人在此谢过。