安裝tensorflow + cuda visual studio integration failed解決方案

前言

在研究一個簡化版本的YouToBe推薦算法的時候發現cpu跑數據確實不夠快。(當時用的學生時代的筆記本,確實慢),前幾周從家裏帶過來上學時自己配的主機。國外服務器淘汰下來的E5處理器,加上當時的甜品顯卡1060。組成了學生時代的興趣。1060也算tensorflow的入門顯卡了,就嘗試裝一下cuda,試試tensorflow-gpu。也是強迫症所致,本人喜歡用最新版的工具和軟件。所以有了下文

1.下載安裝cuda和cudnn

首先去官網 https://developer.nvidia.com/cuda-toolkit-archive 選擇下載了最新的10.1版本cuda從 https://developer.nvidia.com/rdp/cudnn-download 下載最新的cudnn
在這裏插入圖片描述
不論怎麼安裝總是出現這個問題(圖是拿別人的,不復現了)
在這裏插入圖片描述
visual studio integration 安裝總是失敗。用了各種方法,比如其他博客說的
在這裏插入圖片描述
在這裏插入圖片描述
都試了,折騰了一天也沒成功,最後在安裝選項選擇自定義,取消了visual studio integration,確實其他的都安裝成功了。

2.安裝tensorflow

( pip install --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple )自動下載了1.14版本。
嘗試安裝tensorflow時經常因爲socket錯誤導致失敗。嘗試了6-7次,盯着powershell的下載安裝過程,終於成功了。

3.嘗試使用tensorflow-gpu

嘗試運行一個tensorflow程序,報錯ImportError: Could not find ‘cudart64_100.dll’,我查看了
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin 發現有 cudart64_101.dll 這個東西,推測cudart64_100.dll 是10版本的cuda纔有,而10.1版本需要更高版本的tensorflow,嘗試下載高版本的tensorflow。在https://pypi.tuna.tsinghua.edu.cn/simple翻了許久發現最高版本就是1.14 所以高版本的tensorflow是沒戲了,只能嘗試卸載nvidia全部組件,下載10.0的cuda嘗試安裝。

4.安裝cuda V10.0

先清理乾淨系統裏所有的nvidia組件
在這裏插入圖片描述
然後打開cuda V10.0
在這裏插入圖片描述
一路確定就安裝成功了,cnm。
嘗試運行tensorflow程序成功,顯示使用了gpu
在這裏插入圖片描述
查看資源管理器確認了使用gpu
在這裏插入圖片描述
終於成功。
適配版本 cuda 10.0 + cudnn 7.6.2 + tensorflow-gpu 1.14

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章