首先去官网下载cuda安装包, 链接.
我找了半天cuda10.1,没找到,其实这样找:
1、进去之后,直接点击下载 2、点击下图所示:
3、下图中选择一个 4、选择deb(local)安装类型,然后下载。
1、进入安装包所在的目录:
cd 安装包所在的目录2、运行.deb文件,其实就是解压
sudo dpkg -i 你的.deb文件第一次应该会让你安装一个类似: sudo apt-key add /var/cuda-repo-/7fa2af80.pub这样的包,安装即可。然后再次运行上面的命令。
3、更新一下下载源:
sudo apt-get update4、安装cuda
sudo apt-get install cuda5、添加环境变量
sudo gedit ~/.bashrc复制下面的内容到打开的文件的最后面:
export PATH=/usr/local/cuda-10.1/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH这里注意,至于是不是cuda-10.1,去/usr/local/下看看,有的也可能不是。
最后使其生效:
source ~/.bashrc6、重启
reboot命令行输入:
nvidia-smi显示下面这样,说明驱动没问题。 输入下面命令,若显示下图所示,说明安装cuda成功。
nvcc -V查看安装的cuda版本:
cat /usr/local/cuda/version.txt去cudnn的官网下载对应的版本:官网
普通机器 :x86_64architecture 服务器机器 :Power architecture
选择 cuDNN Library for linux
按上图下载下来的是压缩包。要是选择其他的,应该是.deb文件,那就是另外的一种安装方式,我没试过,所以这里也就没写。
将下载下来的压缩包解压(记得解压到home下,右键提取就是解压),当然也可以使用命令来解压:
sudo tar -xvzf cudnn-10.1-linux-x64-v7.6.5.32解压到home下,可以看到cuda文件夹,在当前home目录打开终端,执行下面的命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn*这就已经完成了安装。其实cudnn的安装,就是拷贝头文件和库文件到cuda 对应的文件夹下,并给予权限。
查看cudnn的版本信息,在终端输入:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2