【环境配置】在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
相关推荐
davenian35 分钟前
< 自用文 备份 script :intar.sh> 使用 tar cvfz 命令打包成 .tar.gz 文件来备份多目标(目录,文件)
ubuntu·bash script
西猫雷婶1 小时前
pytorch基本运算-torch.normal()函数生成的随机数据添加噪声
人工智能·pytorch·python·深度学习·学习·线性代数·机器学习
老黄编程1 小时前
禁用内核模块,是否需要执行脚本 $ sudo update-initramfs -u $ sudo update-grub ?
ubuntu
一人の梅雨2 小时前
淘宝店铺全量商品接口深度开发:从分页优化到数据完整性保障
linux·windows·microsoft
Voyager_42 小时前
Linux服务器NAS挂载:Truenas
linux·运维·服务器
Yolo566Q2 小时前
基于PyTorch深度学习遥感影像地物分类与目标检测、分割及遥感影像问题深度学习优化实践技术应用
pytorch·深度学习·分类
liujing102329293 小时前
Day01_Linux移植基础
linux·运维·服务器
小狗爱吃黄桃罐头3 小时前
正点原子【第四期】Linux之驱动开发学习笔记-5.1 设备树下的LED驱动实验
linux·驱动开发·学习
叶凡要飞3 小时前
linux安装google chrome 谷歌浏览器
linux·运维·chrome
清静诗意3 小时前
在 Ubuntu 上通过 Docker 与 Docker Compose 部署项目的完整指南
linux·ubuntu·docker