ubuntu20.4系统,显卡Quadro P620,推荐安装什么版本的tensorflow-gpu,请给出安装详细步骤
时间: 2024-02-23 20:58:58 浏览: 278
对于Quadro P620显卡,建议安装tensorflow-gpu 2.3.0版本,以下是安装步骤:
1. 安装CUDA 10.1和cuDNN 7.6.5
- 首先,需要先安装CUDA 10.1和cuDNN 7.6.5,可以参考NVIDIA的官方文档进行安装。
2. 创建conda虚拟环境
- 打开终端,输入以下命令创建名为tf_gpu的conda虚拟环境:
```
conda create -n tf_gpu python=3.7
```
- 激活虚拟环境:
```
conda activate tf_gpu
```
3. 安装tensorflow-gpu 2.3.0
- 在虚拟环境中,输入以下命令安装tensorflow-gpu 2.3.0:
```
pip install tensorflow-gpu==2.3.0
```
- 安装完成后,可以通过以下命令验证是否成功安装:
```
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
```
4. 验证
- 在终端中输入以下命令,进入python环境:
```
python
```
- 输入以下命令,导入tensorflow-gpu库:
```
import tensorflow as tf
```
- 输入以下命令,查看tensorflow-gpu版本:
```
tf.__version__
```
- 如果输出2.3.0,则表示tensorflow-gpu安装成功。
至此,安装tensorflow-gpu 2.3.0的步骤就完成了。
阅读全文