Window10中成功安装TensorFlow-GPU版本!
我的计算机GPU版本为NVIDIA GeForce RTX 3070
按照网上的安装步骤,我先查看了显卡的版本以及对应的CUDA
桌面鼠标右键->NVIDIA控制面板->帮助->系统信息
一.tensorflow安装
1.安装CUDA
本人计算机CUDA对应的版本为11.2.162
在官网中找到相应的版本号进行下载
https://developer.nvidia.com/cuda-toolkit-archive
下载后安装即可
2.安装cuDNN
从官网中下载对应版本(需要先进行注
原创
2021-07-12 23:43:27 ·
450 阅读 ·
3 评论