1、安裝anaconda
2、新建TensorFlow-gpu環境
3、安裝cuda
(1)查看gpu對應cuda版本:桌面右鍵->NVIDIA控制面板->幫助->系統信息->組件->查看cuda信息(圖1);
(2)下載cuda並安裝:cuda下載地址;(目錄:C:\Program Files\NVIDIA GPU Computing Toolkit)
(3)下載cudnn:cudnn下載地址(需要賬號密碼登錄,qq郵箱、賬號annatu1206、密碼含字母數字特殊字符);
下載完成後不需要安裝,解壓後替換掉cuda9.0根目錄下的文件就行
4、進入TensorFlow-gpu環境,安裝tensorflow-gpu
(1)升級pip工具防止安裝到一半報錯:python -m pip install --upgrade pip
(2)安裝tensorflow-gpu:pip install --ignore-installed --upgrade tensorflow-gpu
遇到的問題:
1、安裝tensorflow-gpu有點問題,import時報錯
原因:安裝的cuda10.1
解決方案:卸載cuda10.1,需保留驅動、系統軟件、vs相關軟件;然後重新安裝cuda10.0
圖1: