矩池云上安装CUDA头文件教程

简介

CUDA头文件(cuda header files)在进行CUDA变成的时候会经常遇到。下面我介绍了一些常用的CUDA头文件及其安装。

英伟达官方源

# 矩池云cuda10及以上
https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

# 矩池云cuda8、cuda9以上
https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/

阿里云镜像源

# 矩池云cuda10及以上
https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/

# 矩池云cuda8、cuda9以上
https://mirrors.aliyun.com/nvidia-cuda/ubuntu1604/x86_64/

apt安装

检查系统版本

source /etc/os-release && echo $VERSION_ID

导入apt国内镜像仓库和其GPG key

# 第一步若输出 16.04 
curl -fsSL https://mirrors.aliyun.com/nvidia-cuda/ubuntu1604/x86_64/3bf863cc.pub | apt-key add -
echo "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1604/x86_64/ /" > /etc/apt/sources.list.d/cuda.list

# 第一步若输出 18.04 
curl -fsSL https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/3bf863cc.pub | apt-key add -
echo "deb https://mirrors.aliyun.com/nvidia-cuda/ubuntu1804/x86_64/ /" > /etc/apt/sources.list.d/cuda.list

案例安装

curand.h

#此版本号10.0 请根据已安装的cuda的版本进行修改
apt-get install cuda-curand-dev-10-0

cusolverDn.h

#此版本号10.0 请根据已安装的cuda的版本进行修改
apt-get install cuda-cusolver-dev-10-0

cublas_v2.h

#此版本号10.0 请根据已安装的cuda的版本进行修改
apt-get install cuda-cublas-dev-10-0

cusparse.h

cusparse.h安装方法来自Facebook Research maskrcnn-benchmark的issues fatal error: cusparse.h: No such file or directory 内。

#此版本号10.0 请根据已安装的cuda的版本进行修改
apt-get install cuda-cusparse-dev-10-0

特别要注意的是cuda11之后的改了包名,需要使用带lib的版本去安装。

apt-get install libcusparse-dev-11-0

deb安装

将对应版本的deb包下载到矩池云上,然后用dpkg命令进行安装。

可以通过浏览器Ctrl + F搜索得到版本信息。

dpkg -i cuda-license-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-cublas-dev-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-cublas-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-cusolver-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-cusolver-dev-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-curand-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-cusparse-10-0_10.0.130-1_amd64.deb
dpkg -i cuda-cusparse-dev-10-0_10.0.130-1_amd64.deb
客服