linux安装tensorflow gpu
时间: 2023-03-20 15:04:40 浏览: 129
安装tensorflow-gpu
5星 · 资源好评率100%
在 Linux 系统上安装 TensorFlow GPU 版本需要先安装 CUDA 和 cuDNN,然后使用 pip 安装 TensorFlow GPU 版本。
1. 安装 CUDA: 下载 CUDA 安装包并运行安装程序。
2. 安装 cuDNN: 下载 cuDNN 安装包并将其解压缩到 CUDA 安装目录下。
3. 安装 TensorFlow GPU: 使用 pip 安装 TensorFlow GPU 版本,例如:pip install tensorflow-gpu
注意:在安装之前请确保你的GPU驱动已经正确安装。
阅读全文