安装适合自己电脑的Cuda和cuDNN

一为什么?

在利用cuda搭建深度学习框架,如pytorch、torchvision等与cuda存在很强的版本对应关系,在下载这些包时一定要查询版本号,其次不同显卡或显卡驱动也与cuda存在对应关心。

二、查询合适的Cuda

进入cmd中,输入以下指令查询

nvidia-smi

安装适合自己电脑的Cuda和cuDNN_第1张图片
查询到这台电脑适合 11.7 版本

三、下载cuda和cudnn

参考链接:https://blog.csdn.net/jhsignal/article/details/111401628

你可能感兴趣的:(python,开发语言,人工智能)