配置环境

安装cudnn

sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb，

sudo dpkg -i libcudnn7-dev_7.5.0.56-1+cuda10.0_amd64.deb

sudo dpkg -i libcudnn7-doc_7.5.0.56-1+cuda10.0_amd64.deb

可以一下子安装三个环境

好像直接安装deb的cudnn.h在usr/include

cuda 版本 
cat /usr/local/cuda/version.txt

cudnn 版本 
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 

pytorch 之前一直安装不上，猜测是因为cudnn没有配置好，关于cudnn 安装问题，我用的是deb 安装后，将usr/include的cudnn放到/usr/local/cuda/include  conda安装数次，conda install pytorch torchvision cudatoolkit=10.0 -c pytorch最终成功。