我要评论
一、安装显卡驱动
1.1 查看显卡型号
lspci | grep -i nvidia
1.2 根据显卡型号选择驱动
英伟达驱动下载：https://www.nvidia.cn/download/find.aspx?lang=cn&qnf=1
根据需求，输入显卡型号、操作系统和语言，选择相应的驱动版本，我选择550.78这个版本。
1.3 获取下载链接
点击下载
点击【同意并开始下载】
复制下载链接：https://cn.download.nvidia.cn/xfree86/linux-x86_64/550.78/nvidia-linux-x86_64-550.78.run
并使用wget在命令行下载：
wget https://cn.download.nvidia.cn/xfree86/linux-x86_64/550.78/nvidia-linux-x86_64-550.78.run
1.4 查看下载的显卡驱动安装文件
ll
1.5 更新软件列表和安装必要软件、依赖
sudo apt-get update
sudo apt-get install g++
sudo apt-get install gcc
sudo apt-get install make
1.6 卸载原有驱动
sudo apt-get remove --purge nvidia*
1.7 禁用默认驱动
在安装nvidia驱动前，禁用系统自带显卡驱动nouveau
使用vim打开和修改文件，也可用gedit
sudo vim /etc/modprobe.d/blacklist.conf
在文件尾增加两行：
blacklist nouveau
options nouveau modeset=0
更新文件
sudo update-initramfs -u
此时必须重启电脑：
sudo reboot
1.8 安装lightdm显示管理器
sudo apt-get install lightdm
1.9 停止显示服务器
sudo telinit 3
1.10 在文本界面中，禁用x-window服务
sudo service gdm3 stop
1.11 安装驱动
# 授予执行权限
sudo chmod 777 nvidia-linux-x86_64-550.78.run
# 执行安装命令
sudo ./nvidia-linux-x86_64-550.78.run
1.12 检测显卡驱动是否安装成功
nvidia-smi
大功告成~
1.12 重启显示服务、恢复图像界面
sudo service gdm3 start
sudo telinit 5
二、安装cuda
2.1 选择合适的cuda版本
我显卡的驱动版本是550.78，可以安装cuda 12版本。可根据自己显卡版本选择合适的cuda版本：查询链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
2.2 下载cuda
下载页面链接：https://developer.nvidia.com/cuda-toolkit-archive
# 使用命令下载文件：
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
# 使用命令查看文件下载：
ll
2.3 安装cuda
sudo sh cuda_12.1.0_530.30.02_linux.run
稍等一会，会出现如下界面，输入：accept即可安装
接着会出现如下界面：
需要注意的是，上述过程我们已在步骤1安装显卡驱动，已无需再装，可通过键盘选择，取消驱动的安装。
再将光标通过键盘移至install，开始安装：
2.4 环境变量配置
以vim方式打开配置文件
sudo vim ~/.bashrc
在文件尾中加入以下两行：
export path="/usr/local/cuda-12.1/bin:$path"
export ld_library_path="/usr/local/cuda-12.1/lib64:$ld_library_path"
更新环境变量
source ~/.bashrc
2.5 检测cuda是否安装成功
nvcc -v
命令行显示如上信息，表明安装成功！大功告成~
三、安装pytorch
3.1 选择合适pytorch版本
链接：https://pytorch.org/get-started/previous-versions/
由于cuda安装的版本是12.1，所以可以用pytorch的v2.2.0版本
3.2 安装pytorch
可用conda安装
conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=12.1 -c pytorch -c nvidia
也可用pip安装
pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121
四、关于安装cudnn（这个坑以后再补）
版本查看页面：https://developer.nvidia.com/cudnn-downloads
发表评论