tensorflowgpu版本的windows安装
首先需要明确需求,即自己需要安装的tensorflow-gpu对应的cuda和cudnn版本。通过该tensorflow官方网站可以查看tensorflow对应的cuda和cudnn版本。
因为我是安装 tensorflow-gpu 的2.2版本,所以从中选择自己需要的cuda和cudnn版本,从而通过cuda官网进行配置
1 配置CUDA与CUDNN
1.1查看本机的CUDA驱动适配版本
桌面右键打开英伟达控制面板,点击帮助->系统信息->组件
从中可以看到我的cuda版本是11.1,则
原创
2020-12-23 14:37:25 ·
253 阅读 ·
0 评论