赞
踩
最近导师让学习ML,由于之前没看过显卡驱动的问题,所以研究了一天,写下来供后人避雷。
一般情况下你直接装的驱动当中是包含有CUDA的,不必去nvidia官网找那个CUDA教程,Deepin和Ubuntu一样,所以不必特别区分下文内容。
首先先打开终端,运行nvidia-smi,如果输出这张图片证明你的CUDA是有的
细节上可能有出入,总之右上角可以显示CUDA版本就可以了。
然后去python里面验证一下CUDA是否可以用
- import torch
- print(torch.cuda.is_available())
输出True的话,这篇文章就可以关掉了,False的话继续往下看。
我也是百度了好久,谷歌也去了好几趟,最后是Deepin官网的一篇教程解决了我的问题,放在这里。
如果认真执行完之后没有解决你的问题,那么你可能需要看一下是不是你的硬件过于老旧或者损坏,或者继续百度下去……
————————————分割线————————————
更新一下,安装CUDA之后,有一部分电脑可能重启没有办法进入图形界面,网上流行的办法是修改grub界面的代码,不过治标不治本,重启之后还是卡在那里。
其实重启之后卡在图形界面外面的时候我们已经进入系统了,这个时候我们修改一下grub的文件重启就好了,首先ctrl+F3进入tty,切换到root下面
vim /boot/grub/grub.cfg
然后找到这一行
在上图148行里,有一个splash quiet,在右面加上和我一样的nouveau.modeset=0,然后wq保存退出。reboot,会发现可以进入图形界面了。
问题原因就是原有的开源N卡驱动和你装的驱动争宠,然后显卡身心疲惫开始摆烂。这样把开源驱动禁掉就好了。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。