ubuntu20.04环境配置
ubuntu20.04环境配置
- NVIDIA驱动安装
- CUDA 12.8版本安装
- conda基础环境配置
NVIDIA驱动安装
- 准备工作
更新系统
确保系统软件包为最新状态:
禁用开源驱动 Nouveausudo apt update && sudo apt upgrade -y
Nouveau 是 Ubuntu 自带的显卡驱动,与 NVIDIA 驱动冲突,需禁用:
在文件末尾添加以下内容:sudo nano /etc/modprobe.d/blacklist.conf
保存后执行:blacklist nouveau options nouveau modeset=0
重启后验证是否禁用成功(无输出即成功):sudo update-initramfs -u sudo reboot
卸载旧版 NVIDIA 驱动(如有)lsmod | grep nouveau
sudo apt purge nvidia-* sudo apt autoremove
- 安装驱动
适配驱动检索
安装apt search nvidia-driver
重启sudo apt install nvidia-driver-570-server-open
验证sudo reboot
结果nvidia-smi
CUDA 12.8版本安装
- 官网选择版本及其他配置https://developer.nvidia.com/cuda-toolkit-archive
运行代码进行安装
wget https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_570.86.10_linux.runsudo sh cuda_12.8.0_570.86.10_linux.run
- 环境配置
接下来配置环境变量,直接在 Linux 命令行输入以下命令:
sudo gited ~/.bashrc
在文件末尾添加 :
export PATH=/usr/local/cuda-12.8/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.8/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda-12,8
更新系统环境:
source ~/.bashrc
sudo ldconfig
验证 CUDA 是否安装成功,输入以下命令:
nvcc -V
conda基础环境配置
pip install torch==2.7.1 torchvision torchaudio -f https://mirrors.aliyun.com/pytorch-wheels/cu128/