ubuntu安装TensorFlow-gpu

在Ubuntu上安装TensorFlow GPU，首先需要确保你的系统已经安装了CUDA和cuDNN。以下是基本步骤：

1. **更新包列表**:

   sudo apt-get update

2. **安装必要的依赖** (如果尚未安装) :

   sudo apt-essential python3-dev python3-pip libopenblas-dev

3. **安装CUDA**:
- 下载CUDA对应版本，可以从NVIDIA官网下载。例如对于CUDA 11.x，访问 https://developer.nvidia.com/cuda-downloads
- 按照指示安装CUDA。

4. **配置环境变量** (替换下面的路径为实际的CUDA安装位置) :

   echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

然后运行 `source ~/.bashrc` 使更改生效。

5. **安装cuDNN**:
- 需要先注册获取cuDNN库。从https://developer.nvidia.com/cudnn 获取并按照提示安装。
- 安装cuDNN到CUDA目录下，通常是 `/usr/local/cuda`。

6. **安装Python 3和pip** (如果还没安装):

   sudo apt-get install python3 python3-pip

7. **通过pip安装TensorFlow-GPU**:

   pip3 install tensorflow-gpu

如果遇到权限问题，可以尝试添加 `--upgrade` 参数升级pip:

   pip3 install --upgrade pip
   pip3 install tensorflow-gpu

8. **验证安装**:
运行 `python3 -c "import tensorflow as tf; print(tf.test.is_gpu_available())"`，如果返回True，则表示TensorFlow成功安装并识别到了GPU。