tensorflow1.15本地配置老版本tensorflow-GPU

Posted _less is more

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow1.15本地配置老版本tensorflow-GPU相关的知识,希望对你有一定的参考价值。

前提:已有anaconda

1、创建虚拟环境

conda create -n tf python=3.7

2、安装cuda和cudnn

进入虚拟环境

conda activate tf

安装cudatoolkit=10.0 cudnn=7.6.5 (稍微大于7.4也可)

conda install -c conda-forge cudatoolkit=10.0 cudnn=7.6.5

3、安装tensorflow1.15

pip install tensorflow-gpu==1.15

Note:
一个tensorflow版本与cuda、cudnn的兼容表

https://www.tensorflow.org/install/source#gpu

以上是关于tensorflow1.15本地配置老版本tensorflow-GPU的主要内容,如果未能解决你的问题,请参考以下文章

无需源码编译 | 基于RTX3090配置tensorflow1.15环境

Tensorflow1 搭建Cuda11

Tensorflow1 搭建Cuda11

Tensorflow1 搭建Cuda11

在python 3.8 / CUDA 11环境下安装tensorflow 1.15

H3C S10512虚拟化配置