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环境