安装cuda和cudnn
安装cuda官方文档:https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
安装cudnn官方文档:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
查看是否有显卡驱动
nvidia-smi
如果有输出即为安装了N卡驱动,并可以看到相应推荐的cuda版本
如果没有安装N卡驱动,可以打开软件更新管理器在驱动选项里进行安装,安装推荐的就好了
或者敲下命令自动安装
sudo ubuntu-drivers autoinstall
检验是否安装完成可以重新执行nvidia-smi
下载相应的cuda和安装
前往官网https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda,需要登陆。
切记一定要是与nvidia-smi显示出来与之对应的,下载run包的话可以比较方便快捷的安装完成,下载deb包可能安装过程还需要另外下载其他东西,不排除会有被墙的可能
run包下载完毕
sudo sh cuda_10.2.89_440.33.01_linux.run
-
点击 continue
-
输入 accept