1. 程式人生 > >windows10+VS+CUDA+cuDNN+TensorFlow-gpu環境搭建(問題及解決)

windows10+VS+CUDA+cuDNN+TensorFlow-gpu環境搭建(問題及解決)

TensorFlow-gpu環境需要CUDA+cuDNN+python,CUDA又需要VS,所以,,,環境越來越大哈哈。

1、主要環境:
Python 3.6
CUDA9.0
Cudann7.0
Tensorflow-gpu1.5
2、安裝步驟(簡略):
Anaconda prompt(python3)->
conda create -n tensorflow python=3.6
->activate tensorflow-gpu
->pip install --upgrade tensorflow-gpu==1.5 -i https://pypi.tuna.tsinghua.edu.cn/simple


(使用清華的源,速度快很多)

3、使用:
Anaconda prompt(python3)->activate tensorflow-gpu->python(進入python環境)-》import tensorflow as tf
4、報錯找不到安裝檔案:安裝以下
1、Microsoft+Visual+S+2017+Redistributable+Package+v14.13.26020(或者2015版本)
2、cuDNN壓縮包中的三個資料夾:lib、include、bin,全選複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0,並加入到path路徑。