废话不多说,简介什么的我也不打算写了,安装过程只做必要的解释。以下操作均在Huawei Matebook Xpro 2018版上运行,Linux系统是Kali 发行版。默认以root用户进行操作
安装步骤总览
一、 安装显卡驱动
二、 显卡驱动检查
三、 Pytorch安装
四、 代码试运行检查
一、安装显卡驱动
1、更新系统
apt-get update
apt-get upgrade
apt-get dist-upgrade
更新系统为安装显卡驱动做准备。
2、上Nvidia官网查找驱动程序
3、转为命令行模式安装
Alt + Ctrl + F3进入命令行模式。
首先将可能存在的旧版NVDIA驱动卸载。(包括之前安装过失败的情况)
apt-get remove nvidia*
禁用nouveau驱动。
vim etc/modrobe.d/blacklist-nouveau.conf
在文件
blacklist-nouveau.co