总所周知TensorFlow-gpu 需要 cuda cudnn TensorFlow-gpu之间版本互相搭配。
遇到一些问题在csdn上找到了帮助,以下分为推荐安装方法和常见问题两部分。
推荐安装方法
1.先上最新的TensorFlow-gpu版本图
2.首先需要参考显卡所支持的cuda版本,链接
3.选择好搭配的版本,手动安装cuda和cudnn:
个人用pycharm装了许多环境,虚拟的,搭配anaconda的,所以很乱。
4.我就另外的建了一个新的虚拟环境,链接
5.之后是,pycharm中手动下载TensorFlow-gpu
(虽然里面是链接中是pytorch,但换成pip install tensorflow-gpu==2.6.0,同样适用)
常见问题:
这个博客里面提到的问题都非常实用,有帮助,链接
随后附上测试代码截图和成功截图。