linux安装tensorflow-gpu
时间: 2023-06-05 10:47:52 浏览: 176
安装tensorflow-gpu
5星 · 资源好评率100%
在安装 TensorFlow-GPU 之前,需要先安装 NVIDIA 驱动、CUDA 和 cuDNN。
1. 安装 NVIDIA 驱动。
2. 安装 CUDA Toolkit。
3. 安装 cuDNN。
之后,在终端中输入以下命令安装 TensorFlow-GPU:
```
pip install tensorflow-gpu
```
安装完成后,可以在 python 中使用以下代码测试是否安装成功:
```
import tensorflow as tf
print(tf.reduce_sum(tf.random.normal([1000, 1000])))
```
如果没有错误输出,说明 TensorFlow-GPU 安装成功。
阅读全文