顯卡設備:英偉達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