一、CUDA安装
./cuda_10.0.130_410.48_linux
安装后会有如下提示:
查看CUDA的版本号
二、CUDNN安装
安装命令
> tar xvzf cudnn-10.0-linux-x64-v7.5.0.56.tgz
> cp cuda/include/cudnn.h /usr/local/cuda/include/
> cp cuda/lib64/* /usr/local/cuda/lib64/
> ldconfig /usr/local/cuda
查看CUDNN版本
三、GPU驱动安装
./NVIDIA-Linux-x86_64-410.57.run
验证方法
执行nvidia-smi,如果一切正常会列出当前机器上的显卡。
如果发现nvidia-smi执行很慢,可以执行下nvidia-persistenced --persistence-mode,再执行nvidia-smi就会变快很多。