RedHat配置GPU计算环境

最新推荐文章于 2024-07-16 17:20:53 发布

·清尘·

最新推荐文章于 2024-07-16 17:20:53 发布

阅读量1.8k

点赞数

本文链接：https://blog.csdn.net/u012969412/article/details/77043353

版权

第一步:安装nvidia显卡驱动

安装文件下载网址：http://www.nvidia.cn/Download/index.aspx?lang=cn

选择对应的显卡型号:down

$ wget -r -O xxxx.rpm  "download_url"

指令集如下：

Once you accept the download please follow the steps listed below 
i) `rpm -i nvidia-diag-driver-local-repo-rhel7-384.59-1.0-1.x86_64.rpm' 
ii) `yum clean all`
iii) `yum install cuda-drivers` 
iv) `reboot`

第二步：安装cuda-8.0环境

下载地址： https://developer.nvidia.com/cuda-downloads

指令集如下(法1，在我的服务器上安装不到目录/uer/local目录)：

Installation Instructions:
`sudo rpm -i cuda-repo-rhel7-8-0-local-ga2-8.0.61-1.x86_64.rpm` #注意对应相对的操作系统
`sudo yum clean all`
`sudo yum install cuda`

指令集如下（法2）：

$ chmod 755 cuda_8.0.61_375.26_linux.run
$ ./cuda_8.0.61_375.26_linux.run

配置cuda-8.0的环境变量：

$ vim /etc/profile

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

$ source /etc/profile

第三步：安装cudnn-V5+到cuda-8.0

到官网下载cudnn 安装包： https://developer.nvidia.com/cudnn

解压并将内容copy到/usr/local/cuda-8.0/include和lib64目录中：

sudo tar xvzf cudnn-8.0-linux-x64-v5.1.tgz  
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include  
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64  
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*   #分配包的权限