tensoeflow2.12.0安装cuda详细步骤

以下是在 Ubuntu 18.04 系统中安装 TensorFlow 2.1.2 和 CUDA 10.1 的详细步骤：

1. 安装 NVIDIA 驱动程序

首先，需要安装 NVIDIA 驱动程序。可以使用以下命令来查看系统上安装的 NVIDIA 驱动程序版本：

nvidia-smi

如果该命令不存在，则需要手动安装 NVIDIA 驱动程序。

2. 安装 CUDA Toolkit

下载 CUDA Toolkit 10.1 的安装包并安装。可以从 NVIDIA 官方网站上下载安装包：

https://developer.nvidia.com/cuda-10.1-download-archive-base

安装时需要选择自定义安装，以便安装 CUDA Toolkit、CUDA Tools 和 CUDA Samples。

安装后，需要将 CUDA 相关的路径添加到环境变量中。可以使用以下命令找到 CUDA 安装路径：

sudo find / -name cuda

找到路径后，将以下内容添加到 ~/.bashrc 文件的末尾：

export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64

然后运行以下命令使环境变量生效：

source ~/.bashrc

3. 安装 cuDNN

下载 cuDNN 7.6.5 for CUDA 10.1 的安装包并安装。可以从 NVIDIA 官方网站上下载安装包：

https://developer.nvidia.com/rdp/cudnn-download

安装时需要将 cuDNN 的库文件复制到 CUDA 的安装路径中：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

4. 安装 TensorFlow

使用 pip 安装 TensorFlow：

pip install tensorflow-gpu==2.1.2

安装完成后，可以使用以下命令测试 TensorFlow 是否能够正常工作：

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

如果没有错误输出，则表示 TensorFlow 已经成功安装。

以上就是在 Ubuntu 18.04 系统中安装 TensorFlow 2.1.2 和 CUDA 10.1 的详细步骤。