服务器安装tensorflow-gpu=2.11.0教程

发布时间 2023-10-22 15:46:33作者: 好问雷

第一步:安装cudatoolkit

  1. 使用nvidia-smi确定适合自己显卡驱动的版本
  2. 使用conda search cudatoolkit获取可用版本
  3. 执行 conda install -c anaconda cudatoolkit=X.X.X
  4. 在~/.bashrc文件中添加 export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/miniconda3_path/envs/tensorflow_env/lib"
  5. 执行 source ~/.bashrc

第二步:安装安装cudnn

  使用conda search cudatoolkit获取可用版本,选择和自己的cuda版本对应的cudnn

第三步:安装TensorRT

  1. here下载 与cuda版本对应的TensorRT,选择tar.gz文件,例如我下载的是TensorRT-8.0.3.4.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz
  2. 执行tar -xzvf TensorRT-8.0.3.4.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz
  3. 在~/.bashrc文件中添加 export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/path/to/TensorRT/lib"
  4. 执行 source ~/.bashrc