環(huán)境:
- Windows 10 x64
- GTX 960
整體流程并不復雜请毛,可分為如下五步:
- 安裝NVidia驅動
- 安裝CUDA 9.0
- 安裝cuDNN 7.x for CUDA 9.0
- 安裝Anaconda
- pip安裝tensorflow-gpu
下面逐一進行記錄
一、安裝NVidia驅動
這里只需要直接百度NVidia部逮,找到NVidia官網(wǎng)院促,下載好驅動安裝就可以了筏养。其實就是在這里搜索對應的顯卡驅動,進行下載常拓。
下載完成后渐溶,重啟一下電腦即可。
二弄抬、安裝CUDA 9.0
本來自己之前安裝的是CUDA 9.1茎辐,但是所有安裝完成后,imort tensorflow時報錯找不到相應的動態(tài)鏈接庫掂恕,細看文件名發(fā)現(xiàn)拖陆,原來tensorflow使用的是CUDA 9.0,所以不得不重新下載安裝懊亡。后來查看官方文檔發(fā)現(xiàn)慕蔚,官方文檔中給出的組件正是 CUDA Toolkit 9.0,嚴格按照官方來斋配,總不會錯孔飒。
去官網(wǎng)下載的時候,默認給出的是最新版9.1的下載地址艰争,注意要選擇Legacy Releases
找到9.0版本坏瞄,這里直接給出地址 https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
下載好 Base Installer 和 Patch 1,先安裝 Base Installer甩卓,然后安裝 Patch 1即可鸠匀。
安裝完成后,打開命令提示符逾柿,輸入 echo %CUDA_PATH%
缀棍,可以看到CUDA_PATH
這個環(huán)境變量已經(jīng)設置好了。TensorFlow官網(wǎng)上要求將這個變量加入到%PATH%
中机错,加入的方式為:
搜索并打開run
> 輸入control sysdm.cpl
> 高級 > 環(huán)境變量 > 選中 path
點擊編輯 > 新建 > 輸入%CUDA_PATH%
> 確定保存
這樣 CUDA 9.0就已經(jīng)安裝好了爬范。
三、安裝cuDNN v7
這里在安裝的時候也是遇到坑的弱匪,就是 https://developer.nvidia.com/ 這個網(wǎng)站總是顯示正在維護青瀑,因而也找不到下載的地方,后來在知乎上找到了解決方法:
nvidia的cudnn為什么不能下載了? - 瞎折騰的一生的回答 - 知乎
里面截圖中給出的鏈接 https://developer.nvidia.com/rdp/cudnn-download 可以使用斥难,進去后注冊登錄枝嘶,同意用戶協(xié)議即可下載。需要注意的是 cuDNN 有一個版本號哑诊,而同一個版本的 cuDNN 適配了不同版本的 CUDA群扶,這需要和自己安裝的CUDA版本對應起來。所以镀裤,需要下載的是 cuDNN v7.1.1 for CUDA 9.0竞阐。
安裝的過程非常簡單,只需要將cuda文件夾下的內容復制到%CUDA_PATH%
下即可.
cuDNN/cuda | %CUDA_PATH% |
---|---|
bin/* | bin/* |
include/* | include/* |
lib/* | lib/* |
四淹禾、安裝Anaconda
推薦清華鏡像源,下載速度有保障茴扁。這里下載的是 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.1.0-Windows-x86_64.exe
五铃岔、安裝tensorflow-gpu
1.配置pip源,使用清華鏡像
https://mirrors.tuna.tsinghua.edu.cn/help/pypi/
2.安裝tensorflow
開始 > Anaconda3 (64-bit) > Anaconda Prompt
輸入
pip install tensorflow-gpu