中国建设银行网站 个人品牌推广渠道

前言

最近公司购买了一张RTX3090的显卡和一台新的服务器，然后对机器的GPU环境进行了安装和配置，然后简单记录一下

环境版本

操作系统：Centos7.8

显卡型号：RTX3090

Python版本：3.7.6

Tensorflow版本：2.5.0

内核：3.10.0-1127.el7.x86_64

操作系统：Centos7.8

显卡型号：RTX3090

Python版本：3.7.6

Tensorflow版本：2.5.0

内核：3.10.0-1127.el7.x86_64

提示：安装环境时一定要按照自己的版本选择对应的安装包，否则过程当中会遇到很多坑。如果使用TensorFlow的版本和Cuda版本不一致，安装成功后则不会调用GPU去跑。对于使用Tensorflow的可以去官网查找对应的包的版本https://tensorflow.google.cn/install/source#gpu

原文链接

安装GPU机器学习环境