安裝Cuda9.0+cudnn7.3.1+tensorflow-gpu1.13.1


我的安裝版本:

  • win10 x64
  • VS2015
  • conda python 3.7
  • 顯卡 GTX 940mx
  • Cuda 9.0
  • cudnn v7.3.1
  • Tensorflow-gpu 1.13.1

 

1.安裝Anaconda

  自動安裝python3.7

2.安裝VS2015

  tensorflow其實是基於VC++2015開發的,所以需要安裝vs2015

  下載地址   https://pan.baidu.com/s/1F7g4sn5qj82RI0syjznFMQ 密碼:fe9f

  https://my.visualstudio.com/Downloads?q=visual%20studio%202015&wt.mc_id=o~msft~vscom~older-downloads

 

3. 安裝顯卡驅動

  去官網找適合自己型號即可

4. 安裝CUDA

  顯卡型號支持:https://developer.nvidia.com/cuda-gpus

  下載安裝CUDA,安裝好之后把CUDA安裝目錄C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下的bin和lib\64添加到Path環境變量中

  在命令行輸入nvcc --version(或者nvcc -V,大寫的V),出現此圖說明成功安裝了vs+cuda+cudnn

  如果之前已安裝高版本顯卡驅動,安裝CUDA時需要自定義並取消最下面的驅動安裝,否則也會安裝失敗。

5. 安裝cuDNN

  下載CuDNN需要注冊賬號

  解壓壓縮包,把壓縮包中bin,include,lib中的文件分別拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0目錄下對應目錄中

6. 拷貝dll

  把C:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v9.0\extras\CUPTI\libx64\cupti64_80.dll拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

7. 進入cmd,安裝gpu版本的TensorFlow

  要注意用conda裝tensorflow時候,會自動下載cuda和cudnn,如下圖所示,我想要cuda9.0,卻被conda自動安裝了10.0;

  而這里用conda自動安裝的版本可能和直接在電腦上安裝的cuda版本不一致,所以在安裝好pip install tensorflow-gpu==1.13.1之后,要重新安裝pip install cudatoolkit==9.0(不用卸載cudatoolkit,因為卸載cudatoolkit會將tensorflow-gpu等相關文件一並卸載,直接裝需要的cudatoolkit版本,就會將原先的版本卸載掉,再安裝新的版本),否則會報內外版本不一致的錯誤。

安裝tensorflow時,直接在命令行輸入安裝速度會很慢,可以嘗試使用國內的鏡像

可以在使用pip的時候在后面加上-i參數,指定pip源

eg: pip install *** -i https://pypi.tuna.tsinghua.edu.cn/simple

  阿里雲 http://mirrors.aliyun.com/pypi/simple/

  中國科技大學 https://pypi.mirrors.ustc.edu.cn/simple/

  豆瓣(douban) http://pypi.douban.com/simple/

  清華大學 https://pypi.tuna.tsinghua.edu.cn/simple/

  中國科學技術大學 http://pypi.mirrors.ustc.edu.cn/simple/

8. 測試是否安裝成功 

import tensorflow as tf
 
x= tf.random_normal((10, 10))
y = tf.random_normal((10, 50))
z = tf.matmul(x, y)
sess = tf.InteractiveSession()
sess.run(z)

print(z)

  

 

 

參考文獻:

【1】【Win7 x64】+【annaconda3】+ 【python3.5.2】+【tensorflow-gpu】 [最終配置 gtx 940mx + Cuda8.0+cudnn v5.1 + tensorflow-gpu1.0.0 ]

【2】Win7 tensorflow gpu安裝

【3】win10 + cuda(v9.0) 安裝TensorFlow-gpu版

【4】win10安裝tensorflow-gpu版安裝連接(visual studio2015+cudn8.0+cudnn6+anaconda3+tensorflow-gpu)

【5】Windows下搭建TensorFlow環境(GPU版本)

【6】Python擴展包的非官方Windows二進制文件

【7】windows 10 64bit+Tensorflow+Keras+VS2015+CUDA8.0+cuDNN v6.0 GPU加速

【8】Win10下Tensorflow(GPU版)安裝趟坑實錄

【9】tensorflow 安裝GPU版本


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM