RTX3090显卡安装tensorflow-gpu 2.4+CUDA11.0+cudnn8.05+py3.7(使用conda安装)
基本环境
Ubuntu 16.04+Anaconda
前期准备
由于实验室电脑有张显卡坏了,买了张Nvidia最新显卡RTX3090,查看TensorFlow和pytorch的官网竟然现在还没有已公布的能匹配的cuda版本(这不应该大家一起坐下来商量好?),于是就开始找找找,终于还是找到了。现在的RTX3090可以配合CUDA11使用。pytorch目前是支持CUDA11.0的,只有TensorFlow目前已公布的最高才CUDA10.0的版本,所以要重装CUDA11。
- 显卡驱动,3090显卡是需要驱动版本455以上的,先看看自己的显卡驱动是什么版本,低了就自己卸载载重装;
- 安装anaconda,我是之前就有就没必要在介绍安装过程了(实在不会就得自己度娘了);
CUDA11.0 安装
首先创建虚拟环境,自行命名,我这是用的python3.7。
conda create -n tensorflow2