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