tensorflow--gpu 安装

本文详细介绍了在Windows 8系统下,使用NVIDIA GTX 960M显卡配置TensorFlow-GPU环境的全过程。从确认显卡类型、安装Python及TensorFlow指定版本,到安装CUDA和cuDNN,解决常见错误,直至成功运行TensorFlow,提供了实用的步骤和注意事项。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

参考https://www.cnblogs.com/pgzhang/p/9146617.html

1 查看自己的显卡是不是独立显卡

2 本机显卡型号 nvidia gtx 960 m,系统为win8

3 安装python3.6.6(64位)

4 安装tensorflow--gpu pip 安装的是1.5版本的,我安装的是1.8 版本的,pip install tensorflow--gpu==1.8.0

注意版本问题:不要随便安装版本,不然会报错说找不到一些dll文件ImportError: Could not find 'cudart64_90.dll'.),

tensorflow-gpu 1.7以及之后的版本要安装 CUDA 8.0 以上的版本,tf 1.7 之下的安装 CUDA 8.0 之下的 ,安装 cuDNN 时要根据 CUDA 的版本来选择

5 安装cuda 在官网下载 cuda9.0

    大概是1.4g的样子,下载之后安装,由于装了360,会有很多的阻止提示,要一直点击允许,不要错过,有些阻止后不知道会不会报错

    安装的时候可以自定义安装的文件,不要太长的路径,因为后续会把cudnn的文件复制过来,可能会报错说没有文件过程,不能复制的问题

    安装玩之后,自定义的cuda的文件后,还会继续安装在c盘的program file文件加里的 有个nvidia gpu computing toolkit 文件

    如果安装正确的话,会有(cuda》v9.0》bin、clude、lib等14的文件)

6 安装之后运行tensorflow 报错说找不到cudnn的dll文件,去官网继续下载cudnn ,版本嗯哼cuda保持一直,对应的9.0

7 讲cudnn的文件解压,有clude,bin,lib/64X,然后把里面的文件分别复制到cuda/v9.0里的相对应的文件里

8 再运行tensorflow 不会再报错

9 如果跑程序的时候报错:if you want to see a list of allocated tensors when oom happend....

减小数值或者bitchsize的大小

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值