查看显卡和cuda是否安装成功

我被这个问题尴尬了很久,

nvidia-smi 查看显卡驱动是否安装成功
nvcc -V 查看cuda是否安装成功,其实也就是cuda是否安装成功

注明一下,以前一直以为要安装cudnn,其实cudnn是个加速包,不安装也没事,照样跑,小白建议别管,因为你用不上。等你用上了,你基本就懂了,另外,安装cuda的时候其实cuda自带cuda驱动,如果nvcc -V不显示,那就请你自己安装一下cuda驱动,就这么简单

记住这两个都需要网上下载对应版本的软件,下载之后正常跟安装其他的软件一样安装就可以,但是cuda安装的时候,一会也会涉及安装pytorch,别去最新官网安装了,因为你已经安装了cuda了,所以找对应版本的安装就可以,网址:https://pytorch.org/get-started/previous-versions/

界面如下所示,这个也是pytorch官网哦
查看显卡和cuda是否安装成功_第1张图片

记得跟python版本要对应,有的时候python版本不对,到时候会影响pytorch的安装

你可能感兴趣的:(GPU,pytorch,深度学习,python)