前言
最近公司购买了一张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