Centos8安装显卡驱动以及Cuda

最近给服务器装了最新的Centos8。本着Dont Repeat Yourself 原则,记录安装显卡驱动和Cuda的过程,以免重复劳动。

找了很多资料,由于Centos8比较新,很多都不起作用了,这里主要参考了linux config下面的一篇博客:https://linuxconfig.org/how-to-install-the-nvidia-drivers-on-centos-8
有空可以翻译一下。

安装之前先yum install 一下gccmakeDevelopment Tools,要注意的是,Cuda安装的是10.2版本,无论是显卡驱动还是Cuda,都要下载.run文件,不要图省事下载rpm,更不要用nvidia-detect。

装完Cuda之后记得添加环境变量,在~/.bashrc里面添加:

export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda

然后source ~/.bashrc

我尝试过给系统换国内源,但是Centos8因为比较新,换阿里源的时候有synchronize问题,目前还没有靠谱的解决方案,所以还是用官方源吧。

最后上几张张安装成功的图,分别是显卡驱动和Cuda,Cudnn没有下好,所以先不安装,以后有需求再装。
Centos8安装显卡驱动以及Cuda_第1张图片
在这里插入图片描述
安装pytorch:pip install torch torchvision
Centos8安装显卡驱动以及Cuda_第2张图片

你可能感兴趣的:(技术笔记)