cuda9.0安装tensorflow-gpu
时间: 2023-04-24 19:02:03 浏览: 207
安装tensorflow-gpu
5星 · 资源好评率100%
1. 首先,确保你的电脑已经安装了CUDA 9.,可以从NVIDIA官网下载安装包并按照提示进行安装。
2. 接下来,你需要安装cuDNN,这是一个加速深度学习的库,也可以从NVIDIA官网下载安装包并按照提示进行安装。
3. 然后,你需要安装tensorflow-gpu,可以通过pip命令进行安装,命令如下:
pip install tensorflow-gpu==1.12.
注意,这里安装的版本号需要与你安装的CUDA和cuDNN版本相对应。
4. 安装完成后,你可以通过import tensorflow来验证是否安装成功。
希望这些步骤能够帮助你成功安装tensorflow-gpu。
阅读全文