1.安装anaconda或者miniconda:
miniconda的安装与相关指令(基于Windows系统)_ant-small的博客-CSDN博客_windows miniconda
2.有需要可以在conda中创建并激活一个虚拟环境,然后来存放接下来要下载的TensorFlow,无所谓的可以直接进行下一步:
conda create -n tf2
conda activate tf2
3.安装TensorFlow:
//CPU和GPU版本已经合在一起了
pip install tensorflow
//如果下载太慢可以使用国内镜像进行下载
pip install tensorflow -i https://pypi.mirrors.ustc.edu.cn/simple/
4.如果要实现tensorflow的GPU支持,那么还要以下的软硬件支持:
硬件要求:支持以下带有GPU的设备,CUDA架构为3.5、5.0、6.0/7.0/7.5/8.0或更高的NVIDA显卡。可以参考这个网站进行查询自己的NVIDA显卡是否支持:CUDA GPUs - Compute Capability | NVIDIA Developer
软件要求:CUDA11.2以及驱动450.80.02 或更高版本。
注:查询本机的CUDA及驱动版本可以在cd到C://Program Files/NVIDA Corporation/NVSMI执行以下命令进行查询:
nvidia-smi
如果版本不够需要进行升级:先更新本机对应的显卡程序,可在该网址进行下载:Official Drivers | NVIDIA