Ubuntu22.04安装paddle GPU版本

确立版本

• 查看官网信息，确立服务版本：https://www.paddlepaddle.org.cn/documentation/docs/zh/2.6/install/pip/linux-pip.html
  

安装CUDA与CUDNN

• 通过nvidia-smi查看当前显卡驱动版本：
  

• 通过https://docs.nvidia.cn/cuda/cuda-toolkit-release-notes/index.html找到与之对应的 CUDA 版本：
  

• 再通过https://developer.nvidia.com/cuda-toolkit-archive/下载该版本的 CUDA ：
  

  wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run
  sudo sh cuda_12.0.0_525.60.13_linux.run --override
  

  报的警告选择 continue ：
  

  根据提示选择安装选项，此处我使用的是lxd容器，之前已经安装过显卡驱动，所以此处选择的不安装，如果是物理机或者第一次配置环境，此处应勾选：
  
  安装完成后提示：
  

• 根据上图提示需要配置环境变量：

  vim  ~/.bashrc
  

  再文件最后加入以下语句（加入的具体内容根据上述输出进行修改）：

  export CUDA_HOME=/usr/local/cuda-12.0
  export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
  export PATH=${CUDA_HOME}/bin:${PATH}
  

  然后使其生效：

  source ~/.bashrc
  

• 可以使用命令 nvcc -V 查看安装的版本信息：
  

• 进入https://developer.nvidia.cn/rdp/cudnn-archive下载对应 CUDA 的 cuDNN ：
  

• 下载好解压：

  tar xvf cudnn-linux-x86_64-8.9.1.23_cuda12-archive.tar.xz
  

• 进入解压后的目录运行以下命令：

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

• 拷贝完成之后，可以使用以下命令查看CUDNN的版本信息：

  cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
  

  
  表示版本信息为8.9.1。

安装paddle

• 查看文档https://www.paddlepaddle.org.cn/documentation/docs/zh/2.6/install/pip/linux-pip.html，使用对应命令安装 paddle：

  python3 -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
  

  

原文地址：https://blog.csdn.net/qq_43650934/article/details/145210343

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！