首先确定想要安装的TensorFlow版本,须知TensorFlow与cuda版本具有对应关系,可以在这里查看从源代码构建 | TensorFlow (google.cn)https://tensorflow.google.cn/install/source#gpu
我用的是CUDA11.2+TensorFlow2.6
那么我们正式开始安装过程:
1、 下载安装cuda
进入NVIDIA官网,初次使用可能需要创建账户CUDA Toolkit 11.6 Update 2 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-downloads
这里可以选择对应电脑系统的安装包,我的电脑是windows10系统
注意注意注意!!!
这里是11.6版本,想要下载旧版本需要在网页下面选择previous cuda releases
下载好后,应该是个exe,双击运行,选择自定义安装,这里的安