首先創(chuàng)建EC2實(shí)例選擇P類型或者G類型
以下是EC2實(shí)例對應(yīng)的N卡產(chǎn)品類型和系列,可以通過這個在官網(wǎng)上找到對應(yīng)的驅(qū)動進(jìn)行手動驅(qū)動安裝
官方鏈接:官方驅(qū)動 | NVIDIA
創(chuàng)建完成登錄遠(yuǎn)程ssh連接至服務(wù)器
連接以后安裝gcc和make
sudo? apt-get install gcc make -y
?更新軟件包緩存并獲取實(shí)例軟件包更新
sudo apt-get update -y
?升級?linux-aws
?程序包以接收最新版本
?sud重啟實(shí)例以加載最新內(nèi)核版本
?重啟實(shí)例以加載最新內(nèi)核版本
sudo reboot
?重新連接實(shí)例
添加NVIDIA PPA
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
為當(dāng)前運(yùn)行的內(nèi)核版本安裝?gcc?編譯器和內(nèi)核標(biāo)頭軟件包
sudo apt-get install -y gcc make linux-headers-$(uname -r)
?禁用 NVIDIA 顯卡的?nouveau
?開源驅(qū)動程序以免與NVIDIA官方驅(qū)動可能不兼容
?將?nouveau
?添加到?/etc/modprobe.d/blacklist.conf
?黑名單文件。
cat << EOF | sudo tee --append /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
EOF
編輯/etc/default/grub 文件并且添加下面的代碼
GRUB_CMDLINE_LINUX="rdblacklist=nouveau"
?重新生成 Grub 配置
sudo update-grub
?更新initramfs,確保新的配置生效
sudo update-initramfs -u
?重啟系統(tǒng)以應(yīng)用更改
sudo reboot
?確認(rèn)Nouveau已被禁用
lsmod | grep nouveau
?如果沒有輸出,表示Nouveau已成功被禁用
安裝推薦的或者特定功能的顯卡驅(qū)動
安裝 ubuntu-drivers-common
sudo apt update
sudo apt install ubuntu-drivers-common
?檢測推薦的驅(qū)動
ubuntu-drivers devices
?可以使用ubuntu-drivers工具自動選擇推薦的驅(qū)動程序
sudo ubuntu-drivers autoinstall
如果需要安裝特定的N卡驅(qū)動可以先搜索可用的N卡包
apt search nvidia-driver
?然后使用apt安裝推薦合適的或者特定的版本
sudo apt install <package-name>
安裝完成以后重啟實(shí)例并且重新連接
?sudo reboot
安裝完成以后輸入命令查看詳細(xì)信息
nvidia-smi
以下是安裝好的效果圖
文章來源:http://www.zghlxwxcb.cn/news/detail-849610.html
如需要手動安裝指定版本或者官方驅(qū)動手動安裝,點(diǎn)贊評論后續(xù)出教程?文章來源地址http://www.zghlxwxcb.cn/news/detail-849610.html
到了這里,關(guān)于使用AWS EC2進(jìn)行AI訓(xùn)練進(jìn)行顯卡驅(qū)動配置的文章就介紹完了。如果您還想了解更多內(nèi)容,請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!