【记录】台式机安装GTX1660驱动过程

原创已于 2025-05-01 15:54:26 修改 · 703 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#学习

于 2025-05-01 12:57:02 首次发布

个人学习记录专栏收录该内容

13 篇文章

订阅专栏

本文仅用于记录自己Ubuntu20系统配置GTX1660显卡驱动的过程。

0. 装机与主板配置

这次装系统搞了很大的坑；BIOS中需要设置 CSM enable然后设置为UEFI mode。否则按照EFI分区安装会找不到grub启动项。
同时，安装时重新格式化完整硬盘，按照原分区装，启动项可能有问题。

1. nvidia驱动

首先需要安装nvida驱动。是否安装通过指令：

nvidia-smi

进行查看。如果没有，则进入ubuntu的software&update，additional drivers里面寻找 535 版本的metapackage，注意不要选server版本。
安装完成后，需要reboot，之后可以通过 smi 指令查看到信息。

2. 安装cuda

首先通过nvidia-smi指令查看给出的驱动版本和右上角的cuda版本，这个版本是当前驱动下所支持的最高的版本。535.180版本驱动支持最高12.2的cuda，因此安装cuda的相关内容。

从 cuda 官方寻找配置过程，https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=deb_local，选择对应的ubuntu版本等。

配置完成后，需要首先添加环境变量，否则找不到cuda。

在 /usr/local 下可以看到cuda的链接和安装目录，将这个路径放到bashrc的PATH中：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.2/lib64
export PATH=/usr/local/cuda-12.2/bin:$PATH

检查：

nvcc --version

输出cuda版本等基本信息。

3. 安装 anaconda

安装时询问是否默认启动anaconda，不管怎样可以在bashrc中增加以下内容，默认启动：

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/home/larry/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/home/larry/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/home/larry/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/home/larry/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
# <<< conda initialize <<<

4. 虚拟环境中安装package

虚拟环境 pytorch，采用python=3.9创建，之后的 cudann, pytorch, tensorboard, scipy-learn, pandas, matplotlib, tqdm,等均采用默认的版本。

pip install nvitop		# install nvidia top

5. python测试代码

import torch
print(torch.__version__)          # 输出 PyTorch 版本
print(torch.cuda.is_available())  # 应返回 True
print(torch.version.cuda)         # 应返回 12.2
print(torch.backends.cudnn.version())  # 输出 cuDNN 版本