养生 装修 购物 美食 感冒 便秘 营销 加盟 小吃 火锅 管理 创业 搭配 减肥 培训 旅游

如何手动搭建使用GPU运行tensorflow的环境

时间:2024-11-15 01:10:54

Tensorflow可以使用gpu多核并行运算,在不安装多余软件的基础上,蛔糯阚熨本经验教你如何在windows上,手动一步步搭建tensorflow在GPU上运行的环境。

工具/原料

联想笔记本82JQ

Windows11

方法/步骤

1、查看gpu最高支持cuda版本号。图示最高支持cuda11.6。

如何手动搭建使用GPU运行tensorflow的环境

2、tensorflow官网install引导殳陌抻泥页,挑选一套适合gpu的配置。图示挑选安装tensorflow2.python3.cudnn8.cuda11.2。

如何手动搭建使用GPU运行tensorflow的环境

3、安装挑选的python版本。

4、nvidia官网下载安装挑选的cuda版本,注意只勾选cuda,并取消勾选visualstudiointegrate。

5、nvidia官网下载挑选的cudnn版本,将解压后的bin、include、lib三个文件夹,复制合并到cuda的安装目录的同名文件夹。

6、安装挑选的tensorflow版本。cmd输入pipinstalltensorflow-gpu==2.9.1,数字为版本号,去掉==2.9.1为默认安装最新版。

7、测试环境。python运行环境下输入:importtensorflowastfpr坡纠课柩int('versi泠贾高框on=%s\nIsgpuavailable?%s'%(tf.__version__,tf.config.list_physical_devices('GPU')))如图返回tensoflow版本号以及使用的GPU信息。

如何手动搭建使用GPU运行tensorflow的环境

8、成功,enjoyit

© 一点知识