2080ti python3.7.0 anaconda配置TensorFlow2.3.1虚拟环境

1、在anaconda中创建好虚拟环境，确定python版本。本次操作使用python3.7.0版本

conda create --name tf231 python==3.7.0

 

输入y确定安装。

如果出现

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <http://mirrors.ustc.edu.

报错，可以重新输入命令，或者换源。

使用命令激活虚拟环境。

conda activate tf231

2、安装cudatoolkit和cudnn。

conda install cudatoolkit=10.1 cudnn=7.6.5

 

3、安装TensorFlow-gpu，使用阿里云镜像加速。

​pip install tensorflow-gpu==2.3.1 -i Simple Index

4、验证是否成功安装TensorFlow，验证gpu是否可用，出现下图证明成功安装。

python
import tensorflow as tf
tf.__version__
print(tf.config.list_physical_devices('GPU'))

验证cuda是否可用

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