商城网站建设清单,邯郸网站建设哪儿好,北京软件外包公司排名,上海装修公司口碑最好的是哪家一、安装驱动
参考这篇文章进行安装Ubuntu安装Nvidia显卡驱动_Kevin__47的博客-CSDN博客 【出现问题】
禁用nouveau后出现黑屏#xff0c;有几行代码#xff0c;断线一直在闪
【解决方法】
1、参考这篇文章Ubuntu20.04安装nvidia显卡驱动并解决重启后黑屏问题_ubuntu安装…
一、安装驱动
参考这篇文章进行安装Ubuntu安装Nvidia显卡驱动_Kevin__47的博客-CSDN博客 【出现问题】
禁用nouveau后出现黑屏有几行代码断线一直在闪
【解决方法】
1、参考这篇文章Ubuntu20.04安装nvidia显卡驱动并解决重启后黑屏问题_ubuntu安装nvidia显卡驱动后黑屏-CSDN博客
在boot 设置中,将安全模式设为disable,保存退出实际我看了安全模式本来就是disable的但是充重启后就能进入ubuntu了。
重新进入后选择ubuntu系统进入。
2、安装原教程中方法验证nouveau是否禁用成功
命令lsmod | grep nouveau
如果没有内容出现则成功。
3、安装驱动
按照原教程运行 NVIDIA-Linux-x86_64-535.104.05.run文件。
报错
提示 缺少cc和gcc模块
【解决】
命令
sudo apt-get install gcc g make
再重新运行run文件最后使用nvidia-smi命令熟悉的界面又回来了
DONE 二、下载CUDA 和CUDNN。
1、版本对应
这里
NVIDIA驱动、CUDA、cuDNN、Torch、Tensorflow对应版本一文搞明白_nvidia驱动对应的cuda版本-CSDN博客
有介绍版本对应但我是在这里Previous PyTorch Versions | PyTorch找到对应关系我pytorch版本是2.0.1cu117,NVIDIA驱动版本是535选择了cuda11.7,cudnn根据cuda版本选择需要登录使用微信登录
【下载链接】
cudaIndex of /compute/cuda/opensource/11.7.0
cudnncuDNN Archive | NVIDIA Developer 三、安装CUDA和CUDNN
1、下载可选多种文件
1、run文件是包含了显卡驱动的文件选用这种方法安装
2deb文件sudo dpkg -i xxxxxx.deb 2、参考基于ubuntu安装cuda,cuDNN_ubuntu安装cuda和cudnn_xyzxyz576的博客-CSDN博客
1安装CUDA 参考ubuntu安装显卡驱动和cuda教程(安装显卡驱动) 进行安装 安装时报错发现没有添加环境变量。添加环境变量
2)安装CUDNN 直接使用命令安装sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.3.28_1.0-1_amd64.deb 使用Ubuntu 安装 GPU 驱动、CUDA、cuDNN以及是否安装成功的检测 - 知乎 的方法测试cuda和cudnn已经成功 四、跑代码
将参数device设置为0YOLOV5代码,即可使用GPU 【总结】
1、先找到显卡驱动、torch、cuda的对应版本再找cudnn的对应版本不然后面会出现一系列问题。
2、安装完驱动要重启。
3、安装cc、gcc
4、关于下载。本案例中cuda使用run文件cudnn使用deb文件
5、安装cuda后要添加环境变量
6、安装完cudnn后可直接去python中测试是否安装成功
