安装cuda和cudnn

1 下载cuda

版本不能过高
一定要和cudnn的最高版本 能对应上,
我第一次没有对应上 ,专门卸载的。

安装完之后验证一下:
安装cuda和cudnn_第1张图片

2

大佬给我的命令行

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

当然需要的话可以把最后换成117 ,代表11.7版本

然后我这就真的可以了呀~

安装cuda和cudnn_第2张图片

你可能感兴趣的:(深度学习理论和实践应用,人工智能应用,人工智能)