在虚拟环境中安装cuda

win10使用anaconda将TensorFlow2.1GPU版安装在虚拟环境中_2020022246的博客-CSDN博客

跑测试代码会显示GPU：false
  测试代码如下：

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)

print('GPU:', tf.test.is_gpu_available())

1.打开Anaconda Prompt

2.建立一个名为tf2.1（该名称可随便取）的环境，使用Python3.7

conda create --name tf2.1 python=3.7

3.进入tf2.1环境

activate tf2.1

4.安装cuda10.1（因为这里安的TensorFlow是2.1版本，对应的cuda版本是10.1，cudnn版本是7.6，如果要安装别的版本的TensorFlow，需安装对应的cuda和cudnn版本。）

conda install cudatoolkit=10.1

5.安装cudnn7.6

conda install cudnn=7.6

6.安装TensorFlow-GPU2.1

pip install tensorflow-gpu==2.1

安装完成就可以使用TensorFlow2.1GPU版本啦！