最近给服务器装了最新的Centos8。本着Dont Repeat Yourself 原则,记录安装显卡驱动和Cuda的过程,以免重复劳动。
找了很多资料,由于Centos8比较新,很多都不起作用了,这里主要参考了linux config下面的一篇博客:https://linuxconfig.org/how-to-install-the-nvidia-drivers-on-centos-8
有空可以翻译一下。
安装之前先yum install 一下gcc
和make
和Development 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没有下好,所以先不安装,以后有需求再装。
安装pytorch:pip install torch torchvision