安装cudann教程

1.首先安装好cuda。
2.下载cudnn的安装,我这里用到的是cudnn v7.0版本,
3.将下载下来的tar文件进行解压:

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

4.接下来,在命令终端执行如下操作:

 cp cuda/include/cudnn.h /usr/local/cuda/include
 cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
 chmod a+r /usr/local/cuda/include/cudnn.h
 chmod /usr/local/cuda/lib64/libcudnn*

然后就大功告成了

你可能感兴趣的:(pytorch)