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())