diff --git a/install-nvidia-smi.md b/install-nvidia-smi.md index 573ae05..e52592c 100644 --- a/install-nvidia-smi.md +++ b/install-nvidia-smi.md @@ -1,3 +1,10 @@ +## 因清华大学开源镜像站 HTTP/403 换了中科大的镜像站 +``` +cat /etc/apt/sources.list +``` + +## 安装了英伟达显卡驱动 + ```shell wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb sudo dpkg -i cuda-keyring_1.1-1_all.deb @@ -10,6 +17,25 @@ sudo apt-get install -y cuda-drivers nvidia-smi ``` +## 安装了CUDA Toolkit +``` +apt install nvidia-cuda-toolkit +``` + ```shell nvcc -V +``` + +## 用 conda 创建了一个 python=3.12 的环境 + +``` +conda create -n vllm-py12 python=3.12 +``` + +``` +conda activate vllm-py12 +``` + +``` +pip install vllm --extra-index-url https://download.pytorch.org/whl/cu128 ``` \ No newline at end of file