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

系统

推荐使用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
相关推荐
怀旧,26 分钟前
【C++】19. 封装红⿊树实现set和map
linux·c++·算法
敲上瘾1 小时前
Docker 存储卷(Volume)核心概念、类型与操作指南
linux·服务器·数据库·docker·容器·架构
神里流~霜灭1 小时前
(C++)数据结构初阶(顺序表的实现)
linux·c语言·数据结构·c++·算法·顺序表·单链表
強云1 小时前
cuda + cudnn + pytorch (win10)
pytorch
禁默2 小时前
已知 inode 号,如何操作文件?Ext 文件系统增删查改底层逻辑拆解
linux·服务器·数据库
许怀楠2 小时前
【Linux】基础指令(下)
linux
咕咚.萌西2 小时前
RISC-V开发环境搭建
linux·硬件架构·risc-v
NiKo_W3 小时前
Linux 初识
linux·运维·服务器
磊灬泽7 小时前
【日常错误】鼠标无反应
linux·windows
抠脚学代码11 小时前
Ubuntu Qt x64平台搭建 arm64 编译套件
数据库·qt·ubuntu