1 安装显卡驱动
2 安装cuda
3 安装cudnn
4 创建虚拟环境,安装tensorflow-gpu(即可下载后本地安装,也可直接pip安装)
完成安装后根据需要修改环境变量中cuda_path,然后激活不同版本的tensorflow虚拟环境
注意事项:
1 不需要安装显卡也能成功是因为window自带显卡驱动,但是该驱动比较旧,所以对于老版本的tensorflow和cuda也许可以,但是新版本的可能不行,建议更新nvidia官方最新驱动
2 如果tensorflow2在import的时候出现不能导入dll的错误,可能是windows版本太老,需要到https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads下载vc_redis进行安装