安装完ubuntu之后,最好立刻去“软件和更新”的“附加驱动”中选择一个驱动进行安装。然后输入“nvidia-smi”,如果打印出显卡的版本等信息就说明驱动安装成功。
成功安装驱动,可以确保“pip3 install torch”时安装的是GPU版本的torch
安装ubuntu时的注意点
最新推荐文章于 2022-11-22 17:34:55 发布
安装完ubuntu之后,最好立刻去“软件和更新”的“附加驱动”中选择一个驱动进行安装。然后输入“nvidia-smi”,如果打印出显卡的版本等信息就说明驱动安装成功。
成功安装驱动,可以确保“pip3 install torch”时安装的是GPU版本的torch