前言

最近公司購(gòu)買了一張RTX3090的顯卡和一臺(tái)新的服務(wù)器，然后對(duì)機(jī)器的GPU環(huán)境進(jìn)行了安裝和配置，然后簡(jiǎn)單記錄一下

環(huán)境版本

操作系統(tǒng)：Centos7.8

顯卡型號(hào)：RTX3090

Python版本：3.7.6

Tensorflow版本：2.5.0

內(nèi)核：3.10.0-1127.el7.x86_64

操作系統(tǒng)：Centos7.8

顯卡型號(hào)：RTX3090

Python版本：3.7.6

Tensorflow版本：2.5.0

內(nèi)核：3.10.0-1127.el7.x86_64

提示：安裝環(huán)境時(shí)一定要按照自己的版本選擇對(duì)應(yīng)的安裝包，否則過程當(dāng)中會(huì)遇到很多坑。如果使用TensorFlow的版本和Cuda版本不一致，安裝成功后則不會(huì)調(diào)用GPU去跑。對(duì)于使用Tensorflow的可以去官網(wǎng)查找對(duì)應(yīng)的包的版本https://tensorflow.google.cn/install/source#gpu

原文鏈接

安裝GPU機(jī)器學(xué)習(xí)環(huán)境