首先安装 cuda 12.1



    Create a file at /etc/modprobe.d/blacklist-nouveau.conf with the following contents:

    blacklist nouveau
    options nouveau modeset=0

    Regenerate the kernel initramfs:

    sudo update-initramfs -u




sudo apt-get install linux-headers-$(uname -r) \
&& sudo apt-key del 7fa2af80 \
&& wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin \
&& sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600 \
&& wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb \
&& sudo dpkg -i cuda-repo-ubuntu2204-12-1-local_12.1.0-530.30.02-1_amd64.deb \
&& sudo cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/ \
&& sudo apt-get update \
&& sudo apt-get -y install cuda


TensorRT 安装命令:

   98  sudo dpkg -i nv-tensorrt-local-repo-ubuntu2204-8.6.1-cuda-12.0_1.0-1_amd64.deb
   99  sudo cp /var/nv-tensorrt-local-repo-ubuntu2204-8.6.1-cuda-12.0/nv-tensorrt-local-42B2FC56-keyring.gpg /usr/share/keyrings/
  100  sudo apt-get update
  101  sudo apt-get install tensorrt

  107  python3 -m pip install numpy
  108  sudo apt-get install python3-libnvinfer-dev
  109  python3 -m pip install protobuf
  110  sudo apt-get install uff-converter-tf
  111  python3 -m pip install numpy onnx
  112  sudo apt-get install onnx-graphsurgeon
  113  dpkg-query -W tensorrt
  114  history

安装 docker engine 并 pull nvidia docker image:

https://docs.docker.com/engine/install/ubuntu/#set-up-the-repository


https://hub.docker.com/r/nvidia/cuda/tags?page=1&name=ubuntu22

docker pull nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐