温馨提示:如果你装CUDA是为了安装GPU版本的pytorch,那么其实不用看我下面关于CUDA安装的内容,因为你使用pytorch官网的安装命令,它会自动帮你安装相应的CUDA,但是cuDNN可能还得你手动安装。
-
安装英伟达专有驱动
利用ubuntu自带的附加驱动软件进行安装,打开附加驱动之后按下图进行操作安装英伟达专有驱动.png
- 测试驱动是否安装成功以及查看驱动版本
打开终端输入nvidia-smi
,查看输出情况。若驱动安装成功,会输出类似下图的结果。
下图中需要注意的有两点:标有1
的方框里面显示的是当前安装的英伟达驱动版本号,标有2
的方框里面显示的是当前驱动版本可以安装的CUDA最高版本号测试驱动安装是否成功以及检查驱动版本.png
- 从英伟达官网下载对应的CUDA安装文件
由上一张图可知,驱动版本号为450.66
,最高支持CUDA 11.0
,