【环境配置】在Ubuntu Server上安装5090 PyTorch环境

风好衣轻2025-06-22 11:47

系统

推荐使用Ubuntu 24.04.2 LTS，22.04可能会有兼容性问题。

driver

配置http/https代理后

shell 复制代码

sudo -E add-apt-repository ppa:graphics-drivers/ppa -y`

使用nvidia-driver-570-open版本：

复制代码

sudo apt install nvidia-driver-570-open

nvidia-cuda-toolkit

shell 复制代码

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-ubuntu2404.pin
sudo mv cuda-ubuntu2404.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda-repo-ubuntu2404-12-8-local_12.8.0-570.86.10-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2404-12-8-local_12.8.0-570.86.10-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2404-12-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-8

就可以找到/usr/local/cuda-12.8/bin/nvcc了

PyTorch

shell 复制代码

pip install torch  --index-url https://download.pytorch.org/whl/cu128