Manjora/Archlinux安装pytorch GPU环境札记

考虑到在ｍａｎｊｏｒａ上配置ｐｙｔｏｒｃｈ的ｇｐｕ开发环境的参考实在太少，同时也希望我所经历的曲折可以对大家有所帮助，于是在此记录下来。
如果有其他问题，可以在评论区留言。

• 首先去ＮＶＩＤＩＡ查询官网了解你的显卡型号是否支持ｃｕｄａ加速.我的型号是GTX 1070。

这里帮某个特殊的可爱的孩子看了一下，ＧＴＸ８５０Ｍ在列表中，无需再次查询。^_
左边是桌面版本的显卡型号，右边是笔记本显卡型号，请对应查找。

• 然后呢，确保你安装的torch是支持ＣＵＤＡ的。一般来说使用pip3(we don’t recommend pip, because it’s so slow.)下载的那个５9０Ｍ左右的包是对的，９０多Ｍ的那个是ＣＰＵ版本，请之前安装过C P U版本的同学将其卸载在重装。

pip3 uninstall torch

另外，从官网给的服务器地址上下载torch网速一般很那突破２００ｋ/ｓ，但是从ｍａｎｊｏｒａ仓库中下载随便就能达到５Ｍ/ｓ。first，去仓库安装一下ＮＶＩＤＩＡ驱动。[it’s okey to choose other drivers once it meet the version demand(>=410) ]

这个过程的长短取决于各自的网速，慢的同学可以先去玩一会泡杯茶了，快的同学继续安装下面的ｐｙｔｏｒｃｈ （torch）。本示例使用的ｃｕｄａ版本是１０，所以在驱动安装的时候选择的是ＮＶＩＤＩＡ　４１０ +，这个是需要匹配才有用的，所以在此声明。
其中包含的大概是这些东西，可以明显看到ｃｕｄａ是ｃｕｄａ１０版本。

这里趁下载的时候做点别的工作。

sudo gedit /etc/modprobe.d/nouveau_blacklist.conf

gedit不是KDE桌面自带的编辑器，但它比较方便复制粘贴，与ｗｉｎ下ｎｏｔｅ编辑器类似，很轻巧，可以先安装一下，sudo pacman -S gedit。如果喜欢且熟悉ｖｉｍ的同学请使用下面的方式打开

sudo vim /etc/modprobe.d/nouveau_blacklist.conf

按ｉ键进入编辑状态。按ＥＳＣ输入：ｗｑ保存退出。（不要少：）
本示例使用ｇｅｄｉｔ（强推这种常见的文件修改使用ｇｅｄｉｔ 或 Kate，不喜欢折腾ｖｉｍ用法）
打开之后添加以下内容：

blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

保存退出。
然后终端继续，删除原来的ＮＶＩＤＩＡ配置，如果原来并没有配置，或者不晓得是否配置的同学，可以走一下流程，确保之前的配置不存在。

cd /etc/X11/
sudo rm xorg.conf
cd

去仓库查一下是否有ｍｅｓａ
有的话进行下一步

sudo pacman -S bumblebee

如果没有ｍｅｓａ则多一步sudo pacman -S mesa即可。
然后将bumblebee添加至用户组。

sudo usermod -a -G bumblebee username

username是你的用户名，当初安装系统设置的用户名（如果不是多用户的话）。比如我的系统用户名为diamonds那么这句应该为

sudo usermod -a -G bumblebee diamonds

然后等之前提到的ＮＶＩＤＩＡ驱动下载安装完再走下一步，否则会出错。

安装完驱动就是时候该启用bumbleed了。

sudo systemctl enable bumblebeed

sudo nvidia-xconfig

如果提示之前没有ｘｃｏｎｆｉｇ，并重被写入，忽略即可，正常。
到这里呢，可以先放下手头的工作（确保下载任务和安装过程已结束），重启机器。
查看一下显卡信息。

nvidia-smi

去终端看看行不行

$ python3
>>>import torch;torch.cuda.is_available()

如果没有成功返回Ｔｒｕｅ的话，
建议再重启一下机子。
重启KDE桌面（如果是gnome桌面，将sddm改成gdm即可，xfce则改成lightdm）
当你走到这一步时，请注意，你的屏幕会立即熄灭。请先看完后续说明在执行一下命令。

sudo systemctl restart sddm

屏幕黑掉之后，如果经过一段时间仍然没有返回图形界面，按ctrl＋alt+F2（如果直接进入桌面可喜可贺），进入系统的命令界面输入用户名和密码
然后呢，关机或重启。（综合来说，gnome最顺利，基本一次到位，KDE还好一些，xfce最麻烦，bugs最多。）
一般来说，到这一步，就能搞定了，如果还是不行，那么本示例已经对各位已经没有帮助了。如果因此搞坏了系统，别打我呀。Ｏ（∩＿∩）Ｏ谢谢。