首先pip install tensorflow==2.1.0
若太慢的话可在上条语句之前更换下载源,使用清华镜像(pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple)
tensorflow默认使用gpu,其基于cuda10.1开发,下载地址
https://developer.nvidia.com/cuda-10.1-download-archive-update2
以及其对应版本的cudnn7.6.4,下载地址
https://developer.nvidia.com/rdp/cudnn-archive
安装完成后,还需要安装vc2015,2017,2019版本,下载地址
https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads
选择自己电脑对应的版本
对应安装完成,重启电脑即可
tensorflow2.1安装配置
最新推荐文章于 2024-05-26 17:25:04 发布