作者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://webptt.com/cn.aspx?n=bbs/DataScience/M.1563712955.A.841.html
1F：推 sxy67230: 重新安装tensorflow 外加重开机，看起来的问题是tensorf 07/21 21:23
2F：→ sxy67230: low 没有侦测到你的cuda。 07/21 21:23
3F：推 sxy67230: 然後cuDNN请确认有没有正确安装，我是在linux环境上装 07/21 21:29
4F：→ sxy67230: 的，所以你找一下window的检查指令 07/21 21:29
5F：→ owem0410: conda安装GPU版本会自己安装cuda-toolkit 所以可以确认 07/22 00:47
6F：→ owem0410: 一下conda package里有没有这一包 07/22 00:47
7F：推 roccqqck: cuda那麽新 tensorflow-gpu要安装高一点版本 07/22 01:24
8F：→ roccqqck: 讲错 旧一点试试 07/22 01:26
9F：→ roccqqck: 试试python3.6 tensorflow1.13 07/22 01:26
10F：推 st1009: 其实可以先用colab 练习 07/22 09:45
11F：→ jackwang01: Windows 极度推荐使用anaconda，现在conda环境有办法 07/22 13:13
12F：→ jackwang01: 直接安装cuda+cudnn ，方便很多 07/22 13:13
13F：→ MGAMEBOY: 感谢大家 刚刚用conda install tensorflow-gpu就可以了 07/22 23:48
14F：推 roccqqck: 原来windows也可以装driver再用conda就好 07/23 13:39
15F：推 erre: mac+外接显卡，十足！ 07/25 05:49
16F：推 chang1248w: tensorflow 支援cudnn10了喔？ 09/05 19:36