RedHat配置GPU计算环境

来源:互联网 发布:全国淘宝店铺有多少个 编辑:程序博客网 时间:2024/05/29 14:53

第一步:安装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*   #分配包的权限