gpu

文章目录

  • 宿主机 安装 driver
  • 测试 nvidia-smi 驱动
  • 使用 gpu-burn 测试 gpu
  • gub-burn 的 dockerfile
  • 赠送的 一些 tf 简单代码

宿主机 安装 driver


# 禁用宿主机 自带 显卡 驱动lsmod | grep nouveaunouveau              1949696  0mxm_wmi                16384  1 nouveauwmi                    32768  2 mxm_wmi,nouveauvideo                  49152  1 nouveaui2c_algo_bit           16384  1 nouveauttm                   106496  2 qxl,nouveaudrm_kms_helper        184320  4 qxl,nouveaudrm                   491520  5 drm_kms_helper,qxl,ttm,nouveau# 首先金庸 nouvean
vi /etc/modprobe.d/blacklist.conf# 在最后一行添加：blacklist nouveaumodprobe_path='/etc/modprobe.d/blacklist.conf'
sed -i "s/blacklist nouveau//g" ${modprobe_path}
echo -e '\nblacklist nouveau' >>  ${modprobe_path}sudo update-initramfs -u# 关闭图形界面
systemctl set-default multi-user.targetrebootlsmod | grep nouveau
# 无输出 代表成功echo "nameserver 114.114.114.114" > /etc/resolv.confsudo sed -i "s@http.*archive.ubuntu@@g" /etc/apt/sources.list
sudo sed -i "s@http.*security.ubuntu@@g" /etc/apt/sources.listapt update -yapt install nvidia-driver-460-server -y# 关闭图形界面
systemctl set-default multi-user.targetdistribution=$(. /etc/os-release;echo $ID$VERSION_ID) \&& cu