cpu训练的时间实在是太长,所以想用gpu训练一下,经过了一天的折腾终于可以用gpu进行训练了,嘿嘿~~~
-
首先先看一下自己电脑的显卡信息,是否存在独显:
-
然后进入NVIDIA控制面板->系统信息->组件 查看可以使用的cuda版本
-
进入官网:https://developer.nvidia.com/cuda-toolkit-archive 找到并下载和显卡信息相匹配的cuda
-
下载完成后按照默认的执行下去(当然也可以修改安装的路径),这里建议取消 VS Integration
-
然后一步步往下执行
-
检查cuda是否安装成功:输入nvcc --version
-
cuda的环境变量在安装时自动配置了,所以不需要我们操心。如果没有配置好,需要手动添加
-
下面安装对应版本的cudnn: https://developer.nvidia.com/cudnn 在这个网站先创建账户,再下载对应版本的cudnn
-
cudnn下载完成后进行解压,里面有三个文件夹
-
将上面的三个文件夹覆盖你之前安装的cuda路径下的相同名称的文件夹(默认安装的cuda和我截图中的路径一样)
-
下面我们使用anaconda安装tensorflow-gpu,【需要特别注意的是:需要确认TF、Python、cuDNN、CUDA,各个版本之间的关系,关系不对应安装过程无警告,但是GPU无法使用。对应的关系可以在这里查到:https://www.tensorflow.org/install/source_windows#gpu】
-
测试结果: