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

 时间:2026-04-21 23:45:15

1、查看gpu最高支持cuda版本号。

图示最高支持cuda 11.6。

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

2、tensorflow官网install引导页,挑选一套适合gpu的配置。

图示挑选安装tensorflow2.9、python3.10、cudnn8.1、cuda11.2。

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

3、安装挑选的python版本。

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

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

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

7、测试环境。python运行环境下输入:

import tensorflow as tf

print('version = %s\nIs gpu available? %s' % (tf.__version__, tf.config.list_physical_devices('GPU')))

如图返回tensoflow版本号以及使用的GPU信息。

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

8、成功,enjoy it

  • Visio如何绘制电量图标#校园分享#
  • 如何使用图像滤波去噪GUI中的加入噪声功能?
  • Matlab2016b安装教程
  • pyCharm如何导入模型库
  • 如何使用Fluent计算十字路口风场(1/3)
  • 热门搜索
    唐太宗属于什么号 布施是什么意思 虚渊玄常被称为什么 饭圈是什么意思 锲而不舍的意思 terrible是什么意思 今年有什么好看的电影 台州有什么好玩的地方 做课件用什么软件 服务器的作用