問題背景:以前用自己的電腦,小小的cpu,跑個mnist,能跑幾十分鐘(衰)。現在有了塊顯卡,雖然也不咋樣,GTX1050,不過好歹也要試試GPU呀。於是參考着各位大佬的教程,一步步摸索,也自己記錄一下,出份教程來。
本文參考鏈接:
https://blog.csdn.net/qq_15969343/article/details/79971469
https://www.cnblogs.com/elroye/p/7864988.html
https://blog.csdn.net/w401229755/article/details/78458014
https://blog.csdn.net/shuiyuejihua/article/details/78738664
https://blog.csdn.net/m0_37160535/article/details/80077903
一、安裝Anaconda以及用Anaconda安裝tf
這個,我之前也寫過一篇博文
https://blog.csdn.net/qq_28123095/article/details/78811598
只不過安裝的是tensorflow-gpu
pip install tensorflow-gpu
另,tensorflow已經支持python3.6了,不用創建3.5的環境來弄了。但我這次用的是3.5
activate tensorflow後,進入python裏:
出現問題:
沒有cuda 9.0,那就裝咯~進入下一步
ps:後來因爲要和其他同學版本一致,卸載了9.0,重裝了8.0,據說卸載cuda很麻煩,但我覺得還好。參考鏈接:
https://blog.csdn.net/shuiyuejihua/article/details/78738664
二、安裝CUDA
到官網,按你自己的需求選擇合適的版本,下載。官網地址:
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=7&target_type=exelocal
這個是9.0的,其他版本的就下其他版本的。
下載好後就安裝嘛。結果就出現問題了:
???
我都讓你裝我C盤了,你還失敗?
然後就百度了一下,度娘說有幾種問題:
1)開了電腦管家、360之類的軟件
2)驅動不匹配,建議用魯大師裝驅動
搞定這兩點,就安裝好啦
對了,我再安裝過程中還出現了黑屏。。。但是我關機重啓後就好了,也不知道是爲什麼???
然後在cmd窗口輸入
nvcc -V
出現cuda版本信息,證明沒問題
三、安裝Cudnn
當然還是官網下載啦,鏈接:
https://developer.nvidia.com/rdp/form/cudnn-download-survey
需要註冊/登錄才行,那就註冊一個賬號咯,然後下載適配CUDA的版本
下載解壓後放到c盤中:
配一下環境變量:
把”c:\cuda\bin”加入path
四、測試
在cmd中輸入下列語句
成功!!!