前言
安装transformers库,运行bert模型时出错,显示torch版本太低,要2.0以上的,所以更新显卡驱动,重新安装pytorch。
1、下载驱动
驱动下载地址:https://www.nvidia.cn/geforce/drivers/
根据自己的需要,选择对应的产品。
使用以下命令可查看服务器对应信息:
# 查看显卡信息
nvidia-smi
# 查看系统信息
lsb_release -a
# 查看系统位数
file /bin/ls
需要注意的是,驱动要和显卡对应上。这里我下载了:NVIDIA-Linux-x86_64-535.171.04.run。