这个部分我是尝试着安装的,没想到一次成功,步骤如下:
先介绍系统、硬件配置
1. 步骤一,CUDA DRIVERS安装
下载地址
我直接下载了最新版,并查看了要求。在页面详细写了:
①支持的显卡;
②支持的最新系统。
需要检查是否匹配。 另外,还标明了CUDA支持的最新版本为CUDA 10.1 Update 1
步骤二,CUDA安装
下载链接。上一步查到了驱动支持的最新cuda版本,下载安装:
安装完成后,配置 CUDA 环境,编辑 ~/.bash_profile 文件, 在命令行运行sudo vim .bash_profile
, 添加下面内容:
export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH="$CUDA_HOME/lib:$CUDA_HOME/extras/CUPTI/lib"
export LD_LIBRARY_PATH=$DYLD_LIBRARY_PATH
export PATH=$DYLD_LIBRARY_PATH:$PATH
export flags="--config=cuda --config=opt"
执行. ~/.bash_profile
刷新。
步骤三, cudnn安装
下载链接。从中选择cuda(我的是10.1)对应的cudnn版本。
解压,然后把解压后的cuda
文件夹复制合并到/usr/local/cuda/
, 当然也可以选择按照官方安装教程
步骤四: 测试
执行如下代码:
cd /usr/local/cuda/samples
sudo make -C 1_Utilities/deviceQuery
./bin/x86_64/darwin/release/deviceQuery
结果如下图出现Result=PASS
,那么CUDA就能够正常工作.
查看CUDNN和cudnn版本:
cat $CUDA_HOME/version.txt # CUDA
cat $CUDA_HOME/include/cudnn.h | grep CUDNN_MAJOR -A 2 #cuDNN