当前位置：网站首页 > 技术教程 > 正文

tensorflow GPU环境安装踩坑日记

csdh11 2025-01-01 19:09 3 浏览

前言：
最近做一个TensorFlow的开源项目，用CPU跑的话，要消耗太多的时间，于是有了这篇配置GPU环境的踩坑日志
分享一些注意的地方

去官网查看经过测试的构建配置

我使用的配置是

Win10+tensorflow2.1+CUDA10.1+cudnn7.6）

地址如下：https://tensorflow.google.cn/install/source#linux

查看构建好的成熟的方案TensorFlow版本和CUDA和cudnn的版本之间的关系

1.安装tensorflow

使用conda创建一个虚拟环境

！！注意！！

pip install -i https://pypi.doubanio.com/simple tensorflow=2.1.0

2.安装cuda10.1

下载地址：

https://developer.nvidia.com/cuda-toolkit-archive

一路next安装完成之后

查看环境变量，默认cuda的环境变量会自动添加

配置path中的环境变量

在终端中测试：

3.cudnn7.6配置文件

文件下载：https://developer.nvidia.com/rdp/cudnn-archive

下载这个文件需要登录NVIDIA官网，我登录了半天登不上去，后面把网页的语言设置成English登上去了

把压缩包解压出来

测试能不能使用GPU

import tensorflow as tf import os os.environ['TF_CPP_MIN_LOG_LEVEL']='2' a = tf.constant(1.) b = tf.constant(2.) print(a+b) print('GPU:', tf.test.is_gpu_available())

GPU环境成功会打印如下

上一篇：小白也能搞定!Windows10上CUDA9.0+CUDNN7.0.5的完美安装教程
下一篇：NVIDIA App上线映众显卡最佳实用工具箱

相关推荐

昔日大尺度女神又一力作，她也太敢了: 今天说的这部剧，光看剧名就已经有点让人瑟瑟发抖了。女主曾经出演过大尺度的影视作品，不过后来凭借《寄生虫》，让她在国际影坛上有了很多好的成绩，一起来欣赏下这部爽剧——...

7个珍藏已久的资源网站，20T硬盘竟不够装，建议低调使用: 7个珍藏已久的资源网站，20T硬盘竟不够装，建议低调使用。哈喽大家好呀~记得之前有给大家分享一些实用网站有收到大家的喜爱，所以今天我又来啦~今天给大家分享7个神仙资源网站，质量高又免费，一起来看看吧。...

新片速递，周末留着慢慢欣赏: 给大家推荐一些新片，利用周末时间可以慢慢观赏。本文图片和内容信息均整理自网络，如有侵权，请联系我们删除。...

这些好评影片，更适合在午夜观看: 浪漫、纯真、怪异……...

“我可以骚，她不能扰”，这短剧太大胆了: 男女主灵魂互换的剧看多了，但是男女身体不变，性别地位互换的倒是少见。最近有一部网剧，设定比较另类，看得人笑喷的同时，也很有讽刺性。另一半的我和你...

猜你爱看:高能不断，全程无尿点: 惊心动魄的冒险之旅，充满奇幻神秘色彩的世界……点开就让人不想暂停的影片，影音先锋今天就给大家推荐两部全程无尿点的电影。...

经典回顾，令人无数次震惊的影片画面: 经典影片大放送，那些不管看多少遍都令人震惊的电影，留下了影史上难忘的珍贵画面。本文图片和内容信息整理自网络，如有侵权，请联系我们删除。...

不良少年和纯情少女谈恋爱，这岛国片嗑疯了: 天气逐渐变冷，急需一些甜甜的剧来温暖心窝。...

少女、骏马、青春，轻松撩动你的心弦: 今天带您走进关于马的世界，探寻少女与马儿的相知相处，看主角如何在过程中找到对人生的追求，变得自信。...

精彩好片推荐，有些需要打码才敢放全: 本文图片和内容信息整理自网络，如有侵权，请联系我们删除。...

今天推荐的这部经典影片，有点“色”~: 在看电影时，还没有深入了解人物和剧情发展之前，影片的画面色彩搭配应该说是最先映入眼帘，给人留下印象的重要部分。色彩的基调与角色特征还有故事背景往往相映衬，关于这一点，下面这部电影就比较经典了。...

R级经典影片，载入影史的绝美造型出处: 本文图片均来源于网络，如有侵权，请联系我们删除。...

经典角色集结，不得不看的年度爽片: 喜欢超级英雄系列电影的，肯定不会错过这部大片。集齐各路经典角色，不管是正派还是反派，都有令影迷喜爱的地方。...

开屏足够抓人眼球，上来就露x+全果的她竟然还活不过一集: 美剧最新播出的剧集出了蛮多高质量的作品，比如前两天说的《后翼弃兵》，以及今天要说的这部高分剧，不仅画面上抓人眼球，剧情设定上也很精彩。...

出演国民顶流MV，岛国清纯女神归来就是禁忌片: 三吉彩花，这个名字大家应该不陌生，很多人知道她就是因为曾经出演过周杰伦MV的女主角。美艳的外表，优越的身材，一看就是模特出身。但很少人知道她是女团的成员，可能更少人知道，她还出演过电影作品。...

一周热门

最近发表

首页
收录
顶部