作者MGAMEBOY (超級潛水艇)
看板DataScience
標題[問題] tensorflow-gpu環境問題
時間Sun Jul 21 20:42:33 2019
作業系統: win7
問題類別: tensorflow-gpu環境設定
使用工具: python3.5(Anaconda3-2019.03),tensorflow-gpu-1.14,
CUDA 10.0,cuDNN 7.4.1
問題內容:
各位版上的大大好
我是個剛接觸這領域的新手
在建立使用顯卡來進行運算的環境上遇到問題
無法以GPU來運算
所以想來請好心的大大指點指點
感謝各位大大
我以指令來確認並沒有出現/device:GPU:0的資訊
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
https://imgur.com/zV16QiE
以下是我的操作流程：
1.更新顯示卡驅動程式
2.安裝Anaconda & Tensorflow-gpu
3.安裝CUDA和cuDNN
4.設定環境變數
在Path新增
D:\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
D:\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
D:\NVIDIA GPU Computing Toolkit\cuDNN\v7.4.1.5\bin
nvcc -V指令結果如下
https://imgur.com/48im3Ad
我有查過tensorflow-gpu需要的CUDA和cuDNN版本需求
https://imgur.com/BonzCfF
自己的顯卡為GV-N660 OC-2GD
雖然很弱但想先試試看
之後再考慮採購新機
顯卡資訊
https://imgur.com/mQpTo6z
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 1.173.213.165 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1563712955.A.841.html
推 sxy67230: 重新安裝tensorflow 外加重開機，看起來的問題是tensorf 07/21 21:23
→ sxy67230: low 沒有偵測到你的cuda。 07/21 21:23
推 sxy67230: 然後cuDNN請確認有沒有正確安裝，我是在linux環境上裝 07/21 21:29
→ sxy67230: 的，所以你找一下window的檢查指令 07/21 21:29
→ owem0410: conda安裝GPU版本會自己安裝cuda-toolkit 所以可以確認 07/22 00:47
→ owem0410: 一下conda package裡有沒有這一包 07/22 00:47
推 roccqqck: cuda那麼新 tensorflow-gpu要安裝高一點版本 07/22 01:24
→ roccqqck: 講錯 舊一點試試 07/22 01:26
→ roccqqck: 試試python3.6 tensorflow1.13 07/22 01:26
推 st1009: 其實可以先用colab 練習 07/22 09:45
→ jackwang01: Windows 極度推薦使用anaconda，現在conda環境有辦法 07/22 13:13
→ jackwang01: 直接安裝cuda+cudnn ，方便很多 07/22 13:13
→ MGAMEBOY: 感謝大家 剛剛用conda install tensorflow-gpu就可以了 07/22 23:48
推 roccqqck: 原來windows也可以裝driver再用conda就好 07/23 13:39
推 erre: mac+外接顯卡，十足！ 07/25 05:49
推 chang1248w: tensorflow 支援cudnn10了喔？ 09/05 19:36