上一篇是在Windows下配置的环境,代码太占显存,换一个服务器。呜呜呜呜呜呜,又要配环境!!!!!这一篇在Ubuntu下从头到尾配置。
设备:Ubuntu18.04+2*Titan RTX(24GB)
一、安装NVIDIA驱动(终端出现下图结果表示安装成功)。驱动版本为515.48.07,CUDA版本为 11.7。我装好cudnn后发现tensorflow的gpu版本还不支持cuda11.7!!!啊啊啊啊啊啊啊啊啊啊啊啊啊又是疯掉的一早上。
参考:ubuntu18.04 安装Nvidia驱动的三种方式(必看)_低吟浅笑的博客-CSDN博客_ubuntu18.04安装nvidia显卡驱动
二、查看对应版本!!!!
tensorflow目前最高支持cuda11.2。
1.首先查看NVIDIA对应的CUDA版本。