【Tebsorflow-gpu 1.15 安装】:Anaconda下安装

环境 :python3.6
激活环境:

conda activate tf15
conda install cudatoolkit=10.0

查找cudnn对应版本:

conda search cudnn -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

【Tebsorflow-gpu 1.15 安装】:Anaconda下安装_第1张图片
安装cudnn

conda install cudnn=7.6.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

安装tensorflow

pip install tensorflow-gpu==1.15

进入编译环境测试GPU是否可用:

在这里插入图片描述

import tensorflow as tf
 print(tf.test.is_gpu_available())

【Tebsorflow-gpu 1.15 安装】:Anaconda下安装_第2张图片
安装成功!

你可能感兴趣的:(python)