前言

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

環(huán)境版本

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

顯卡型號：RTX3090

Python版本：3.7.6

Tensorflow版本：2.5.0

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

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

顯卡型號：RTX3090

Python版本：3.7.6

Tensorflow版本：2.5.0

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

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

原文鏈接

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