cuda11.2对应的tensorflow-gpu
时间: 2023-09-19 15:13:26 浏览: 116
Cuda10.0 + Cudnn7.4.1 + tensorflow-gpu==1.14.0
5星 · 资源好评率100%
如果您的GPU驱动支持CUDA 11.2,您可以安装TensorFlow-GPU 2.5.0或更高版本来使用CUDA 11.2。您可以使用以下命令安装最新版本的TensorFlow-GPU:
```
pip install tensorflow-gpu
```
请注意,您需要安装适用于CUDA 11.2的cuDNN库。您可以从NVIDIA的官方网站下载并安装适用于CUDA 11.2的cuDNN库。
阅读全文