一:系統(tǒng)與環(huán)境
系統(tǒng):windows10
GPU:NVIDIA GeForce GTX TITAN X
python: Anaconda, python 3.7
需要安裝的以來(lái):Visual Stutio, CUDA, cuDNN
首先要確定要安裝的Tensorflow的版本以及他的對(duì)應(yīng)的各種以來(lái)的版本
首先要確定要安裝的Tensorflow的版本以及他的對(duì)應(yīng)的各種以來(lái)的版本
首先要確定要安裝的Tensorflow的版本以及他的對(duì)應(yīng)的各種以來(lái)的版本
重要的事情當(dāng)然說(shuō)三遍!
首先在 確定版本鏈接 這里確定你需要的tensorflow版本匾效,以及其依賴(lài)對(duì)應(yīng)的版本:
- MSVC表示Visual Stutio
二:安裝步驟
1. 安裝Visual Stutio 2017
Visual Studio 這里可以選擇版本炮姨,然后安裝Community就可以吴超。
2. 安裝CUDA 10.0
按照版本下載后,然后一路安裝:
注意: 使用默認(rèn)路徑烤蜕!
命令行輸入指令測(cè)試安裝是否成功:nvcc -V
3. 安裝cuDNN 7.4
3.1從 這里Archive 中選擇對(duì)應(yīng)的版本。下載之后,解壓文件夾吮廉,將bin/,include/,lib/三個(gè)文件夾直接拷到CUDA的安裝目錄中,我這里是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
3.2 將CUDA的目錄加到環(huán)境變量中(CUDA安裝成功之后畸肆,他會(huì)自動(dòng)加進(jìn)去一部分宦芦,這里建議把列出來(lái)的都加進(jìn)去):
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp;
4. 創(chuàng)建tf-gpu環(huán)境,安裝tensorflow-gpu 1.14,使用清華大學(xué)的pip源
> conda create -n tf-14 pip python=3.7 #新建tf-14環(huán)境
> acitvate tf-14 #激活環(huán)境
> pip install --ignore-installed --upgrade tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple # pip安裝tf-gpu版本
5. 測(cè)試一下
進(jìn)入python模式,輸入import tensorflow as tf
ps:這里的FutureWarning是與numpy的版本不符導(dǎo)致的轴脐,可以忽略调卑。