ubuntu18.04安装tensorflowgpu
时间: 2023-04-25 19:01:12 浏览: 162
安装tensorflow-gpu
5星 · 资源好评率100%
安装 TensorFlow GPU 版本需要在 Ubuntu 18.04 上执行以下步骤:
1. 安装 Nvidia 显卡驱动程序。可以使用以下命令检查系统中已安装的显卡驱动程序版本:
```
$ ubuntu-drivers devices
```
然后使用以下命令安装推荐的驱动程序版本:
```
$ sudo ubuntu-drivers autoinstall
```
2. 安装 CUDA。可以从 Nvidia 的官网上下载适合自己系统的 CUDA 版本,然后执行安装程序进行安装。
3. 安装 cuDNN。在 Nvidia 的官网上下载适合自己系统和 CUDA 版本的 cuDNN 版本,然后将其解压到安装目录中。
4. 安装 TensorFlow。可以使用 pip 命令安装 TensorFlow GPU 版本:
```
$ pip install tensorflow-gpu
```
安装完成后,就可以在 Python 中使用 TensorFlow GPU 版本了。
阅读全文