Tensorflow-gpu环境搭建
显卡设备:英伟达1060
系统:Windows10
python版本:3.7.4
CUDA:cuda_10.0.130_411.31_win10
cudnn:cudnn-10.0-windows10-x64-v7.6.5.32
tensorflow:gpu1.14
一 安装python3.7.4
https://www.python.org/downloads/windows/ 下载安装
二 安装cuda:
1.首先更新显卡驱动;
2.桌面右键-NVIDIA控制面板-帮助-系统信息-组件,可以查看当前设备支持的最高cuda版本。
我更新驱动之前支持最高到8,更新后就支持10版本了。
3.下载cuda
https://developer.nvidia.com/cuda-downloads
安装。一直下一步就行。选择组件的时候可以不选择显卡驱动,如果你的驱动比较新。
4.环境变量
三 cudnn安装
1 下载https://developer.nvidia.com/rdp/cudnn-archive cuda和cudnn有对应版本关系。
2.解压文件夹,将解压后的文件夹下的文件拷贝到cuda安装目录下,与之相对应的文件夹下。
解压后其实是三个文件夹,合并到cuda的安装目录中就可以了。
默认路径是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 找不到了可以类比找一下
四 安装tensorflow-gpu
cmd中直接运行
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14