安装ubuntu时的注意点

安装完ubuntu之后，最好立刻去“软件和更新”的“附加驱动”中选择一个驱动进行安装。然后输入“nvidia-smi”，如果打印出显卡的版本等信息就说明驱动安装成功。
成功安装驱动，可以确保“pip3 install torch”时安装的是GPU版本的torch