cuda tensorflow版本对应_Windows10下安装tensorflow-gpu(2.2.0)安装教程(避坑+保姆式教学)...
一、下載和安裝CUDA
(1)查看電腦的顯卡類型和顯卡驅(qū)動(dòng)版本
(2)查詢CUDA版本與顯卡驅(qū)動(dòng)版本對(duì)照表
網(wǎng)址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-resolved-issues
(3)(選做)根據(jù)上表,若需要更新顯卡驅(qū)動(dòng),可前往NVIDIA驅(qū)動(dòng)程序下載
網(wǎng)址:https://www.nvidia.cn/Download/index.aspx?lang=cn
(4)下載你想要的CUDA版本,我在這里選擇了10.2版本
網(wǎng)址:https://developer.nvidia.com/cuda-toolkit-archive
根據(jù)自己的電腦選擇對(duì)應(yīng)的版本,切記選擇本地安裝exe(local),點(diǎn)擊Download進(jìn)行下載。
(5)安裝CUDA
1.在你想要安裝的目錄上創(chuàng)建名為CUDA-temp和CUDA的文件夾。
2.將臨時(shí)解壓路徑設(shè)置為剛創(chuàng)建的CUDA-temp文件夾的路徑。
3.選擇自定義安裝
4.在自定義安裝選項(xiàng)中只選擇紅框中的選項(xiàng),其它選項(xiàng)不必勾選
5.將安裝位置修改為CUDA文件夾中的對(duì)應(yīng)位置(也可保持默認(rèn)設(shè)置)
6.安裝完成
(6)配置環(huán)境變量
1.按Win+R輸入control進(jìn)行控制面板,選擇系統(tǒng)和安全,再選擇系統(tǒng),之后點(diǎn)擊高級(jí)系統(tǒng)設(shè)置,點(diǎn)擊環(huán)境變量。
2.將下面四個(gè)路徑加入到系統(tǒng)變量的Path中,注意要換成自己的安裝路徑。
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2bin
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libx64
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libnvvp
(7)檢查CUDA是否安裝成功
按Win+R輸入cmd,在控制臺(tái)輸入nvcc -V,返回以下信息,說(shuō)明CUDA安裝成功
二、下載和配置cuDNN
下載網(wǎng)址:https://developer.nvidia.com/rdp/cudnn-archive
(1)選擇對(duì)應(yīng)CUDA的cnDNN的版本(這里沒(méi)有對(duì)應(yīng)10.2的版本,因?yàn)檫x擇了對(duì)應(yīng)10.1的版本,不影響后續(xù)安裝)
(2)下載前需要注冊(cè),注冊(cè)后即可進(jìn)行下載
(3)下載完成后,將cuDNN解壓到CUDA同一個(gè)文件夾,方便后續(xù)操作
(4)將cnDNN三個(gè)文件夾下的文件分別復(fù)制到CUDA安裝目錄下的對(duì)應(yīng)文件夾里即可
三、創(chuàng)建Anaconda虛擬環(huán)境
打開Anaconda的控制臺(tái)(Anaconda的安裝和使用大家可以參考其它教程,在這里就不贅述了),創(chuàng)建一個(gè)新的虛擬環(huán)境。
conda create -n DL python=3.6
四、安裝tensorflow-gpu2.2.0
(1)啟動(dòng)剛創(chuàng)建的虛擬環(huán)境
(2)安裝tensorflow-gpu 2.2.0(這里使用清華源)
pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
(3)添加缺少的cudart64_101.dll文件
安裝完成后,若直接運(yùn)行,會(huì)報(bào)錯(cuò)缺少cudart64_101.dll。
解決方案:將缺少的cudart64_101.dll文件復(fù)制到CUDA安裝的NVIDIA GPU Computing ToolkitCUDAv10.2bin目錄下即可。
(這個(gè)dll可以在網(wǎng)上找到,同樣也可以私信我!)
(4)大功告成!檢查tensorflow是否能夠使用gpu運(yùn)行
t tensorflow as tf tf.test.is_gpu_available() tf.config.list_physical_devices('GPU') tf.test.gpu_device_name()寫在最后
文章會(huì)同步發(fā)布在CSDN和知乎,歡迎大家一起交流,共同學(xué)習(xí)進(jìn)步!
CSDN傳送門
https://me.csdn.net/xylbill97?me.csdn.net知乎傳送門
xylbill?www.zhihu.com總結(jié)
以上是生活随笔為你收集整理的cuda tensorflow版本对应_Windows10下安装tensorflow-gpu(2.2.0)安装教程(避坑+保姆式教学)...的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: PID控制器改进笔记之一:改进PID控制
- 下一篇: java信息管理系统总结_java实现科