Ubuntu18.04 + RTX3090安装tensorflow-GPU1.15.4

因为RTX3090只支持cuda11.0+的版本,无法配置tensorflow1.×环境。好在NVIDIA一直在维护一个1.15版本的nvidia-tensorflow可以使用在30系显卡上。安装步骤如下:

准备工作

anaconda(推荐),当前Driver Version: 470.161.03(参考)

1. conda创建虚拟环境

conda create --name tf1.15 python=3.6
conda activate tf1.15

2. 安装tensorflow wheel的索引

pip install nvidia-pyindex

3. 安装tensorflow-GPU

pip install nvidia-tensorflow==1.15.4

亲测1.15.4可用,若不指定版本默认安装1.15.5,GPU不能正常运行。
上面2个命令行可能无法一次性安装成功的,可以多试几次。

4. 测试tensorflow-GPU是否安装成功

依次输入pythonimport tensorflow as tftf.__version__tf.test.is_gpu_available(),查看tensorflow版本并测试GPU能否正常运行。
Ubuntu18.04 + RTX3090安装tensorflow-GPU1.15.4_第1张图片

参考

【1】无需源码编译 | 基于RTX3090配置tensorflow1.15环境
【2】RTX3080+Ubuntu18.04+cuda11.1+cudnn8.0.4+TensorFlow1.15.4+PyTorch1.7.0环境配置
【3】https://developer.nvidia.com/blog/accelerating-tensorflow-on-a100-gpus/

你可能感兴趣的:(tensorflow,ubuntu,nvidia)