如何手动搭建使用GPU运行tensorflow的环境
1、查看gpu最高支持cuda版本号。图示最高支持cuda 11.6。
2、tensorflow官网install引导殳陌抻泥页,挑选一套适合gpu的配置。图示挑选安装tensorflow2.9、python3.10、cudnn8.1、cuda11.2。
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 tfpr坡纠课柩int('version = %s\nIs gpu available? %s' % (tf.__version__, tf.config.list_physical_devices('GPU')))如图返回tensoflow版本号以及使用的GPU信息。
8、成功,enjoy it
声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
阅读量:22
阅读量:38
阅读量:88
阅读量:21
阅读量:52